1. 程式人生 > >yolo_v2入門實現目標檢測

yolo_v2入門實現目標檢測

python開發環境,安裝了tensorflow-gpukeras和其他常用機器學習庫。該過程使用yolo_v2進行測試。

獲取YAD2K原始碼

yolo有專有的深度學習網路darknet,YAD2K是python使用yolo_v2網路的一種介面方法。
克隆或下載https://github.com/allanzelener/YAD2K原始碼。

獲取模型、權重檔案

模型轉換

在YAD2K原始碼資料夾中執行如下命令:

python .\yad2k.py .\my_config\yolov2.cfg .\my_config\yolov2.weights .\model_data\yolo
.h5

其中my_config為步驟2的兩個檔案所在資料夾。該操作將 Darknet YOLO_v2 model轉換為了Keras model,在model_data資料夾生成了兩個檔案:
這裡寫圖片描述

執行測試

python .\test_yolo.py .\model_data\yolo.h5

該操作使用images資料夾中的圖片,生成圖片到images/out資料夾。

這裡寫圖片描述
這裡寫圖片描述

該過程精簡修正程式碼(不是我的哈哈,0.0)在GitHub上。