使用YOLOV3達成物件偵測功能(使用IDE輸出)

 1.研究動機

在人工智慧概論中,老師介紹了數種的深度學習算法,例如VGG16ResNet,物件偵測的算法Mask-RcnnYolo等,我對物件偵測非常的有興趣,在網路上搜尋作法時,發覺Yolo是一個包裝好的套件,舉凡預測影片圖片,都需要使用Yolo的指令,因此我想將Yolo的功能,搬移到用一般的IDE也可以使用Yolo

2.研究步驟

2.1獲得yolo之預測輸出

Yolo演算法的darknet資料夾中,有一個名為darknet.pypython檔案,為作者讓使用者自行使用的python接口,不過由於該檔案預測目標的座標是輸出錯誤的,因此需要先更改檔案使其預測正確,以下為使用更改後的darknet.py預測輸出結果之截圖,該檔案會回傳預測之陣列,內含預測之類別以及該類別之座標

3.實驗結果

    以下為單張圖片之範例,獲得座標之後即可依照座標使用opencv函式庫標出方框,這樣的輸出結果跟yolo指令的輸出結果一模一樣,但是藉由獲得預測的輸出座標,可以產生更多的運用。

圖一Yolo執行畫面程式動態影片

4.結論

在把yolo搬移到IDE上執行後,更多可能的應用例如:即時影片馬賽克、計算物體的出現次數等等,相信在未來需要使用到目標檢測的算法時,可以更為得心應手。


留言

這個網誌中的熱門文章

以dlib實現人臉辨識打卡系統

以python讀取yolo的輸出-darknet.py

使用Python達成影像形態學處理(不使用Opencv函式庫)