yolo_v2入門實現目標檢測
阿新 • • 發佈:2019-02-09
python開發環境,安裝了tensorflow-gpu和keras和其他常用機器學習庫。該過程使用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上。