PythonからのYOLO v3 導入

前提

https://github.com/arunponnusamy/object-detection-opencv
に従う。

git clone https://github.com/arunponnusamy/object-detection-opencv.git

本家のナイーブな方法、keras-yolov3も試したが、失敗。

要件

• Python 3 (2は未検証)
• opencv
• numpy

pip install opencv-pythonしておく

学習済みのweightsのダウンロード

$ wget https://pjreddie.com/media/files/yolov3.weights
さっき落としてきたobject-detecction-opencvに入れる

テストコードの実行

python yolo_opencv.py --image dog.jpg --config yolov3.cfg --weights yolov3.weights --classes yolov3.txt

実行画面

こんな感じで物体認識されている。