1. 程式人生 > >tensorflow + python + keras-yolov3目標檢測

tensorflow + python + keras-yolov3目標檢測

keras-yolo3原始碼地址:

要跑通這個專案,首先,在官網下載yolo模型引數,然後把yolov3.cfg yolov3.weights轉換成yolo.h5檔案放在model_data/下,然後執行convert.py 第一個引數為yolov3.cfg yolov3.weights模型引數檔案,第二個引數為轉換後的模型檔案的儲存路徑model_data/yolo.h5

轉換後即可執行yolo_video.py

引數:

--image  對圖片進行目標檢測

--input [視訊地址]  對視訊進行目標檢測 視訊格式為mp4或者avi

圖片執行效果如下: