YOLO (You Only Look Once) 是關於物件偵測的類神經網路演算法,
作者 Joseph Redmon
以 darknet
架構
實作,輕量、依賴少、演算法高效率,在工業應用領域很有價值,例如行人偵測、工業影像偵測等等。官網寫的非常詳盡,照著操作便能完成 Yolo 初步的 detection 和 training
,
但
這裡只測試另一款使用 keras 架構實作的 keras-yolo3
的操作步驟。
操作步驟:
下載 YOLO 原始碼, 這裡只用到它的 data 而不進行編譯 :
git clone https://github.com/pjreddie/darknet
下載 keras-yolo3 與預訓練的 weight:
wget https://pjreddie.com/media/files/yolov3.weights
python convert.py yolov3.cfg yolov3.weights model_data/yolo.h5
將 darknet 目錄下的 data 複製至 keras-yolo3\data
測試影像辨識 :
python yolo_video.py --image
此時會詢問影像檔名稱, 輸入 data\dog.jpg 或 data目錄下的任何影像檔, 即可看到輸出畫面:
測試視訊辨識, 須自備視訊檔 my_video.avi:
python yolo_video.py --input my_video.avi
沒有留言:
張貼留言