2018年12月9日 星期日

keras-yolo3 測試

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
     
     
     

    沒有留言:

    張貼留言