使用YOLOV4 輕鬆進行開關開閉狀態檢測

使用YOLOV4 輕鬆進行開關開閉檢測

最近接到了一個開關開閉狀態檢測的項目,準確率要求很高,要達到99.9%以上,所以順道試驗一下yolov4的性能
yoloV4的訓練過程跟yolov3訓練並無差異,我把yolov4的git地址和yolov3訓練的參考的博客地址放在這裏
訓練yolov3的過程
yolov4的代碼地址
如果需要yolov4的權重文件和主幹網絡的權重文件,請在這裏下載

我訓練集使用了450張圖片,測試效果時使用了1000個圖片
訓練結束後,跟以前一樣用opencv直接調用cfg文件和權重文件發現報錯了,無法直接調用
這裏我使用的是pytorch調用darknet的方式來調用模型進行測試:
測試算法代碼的地址在這裏:
https://github.com/gitTongzhen/pytorch_yolov4_detect

yolov4的準確率滿足要求,在GPU的實時性還行,大致爲1080TI的服務器50ms一張圖片,但是單位過於貧窮,沒有gpu的服務器,項目需要部署到CPU端,實時性要求比較高,目前測試的i5雙核的每張圖片需要5000ms,和現在還不理解爲什麼cpu端和gpu端的性能會差這麼多,有大佬看到的話,希望能夠賜教。
目前只能放棄yolov4部署到cpu端的想法
效果圖如下圖所示
效果圖

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章