目的

Macでyolo v3を動かして画像認識した際の備忘録です

YOLO: Real-Time Object Detection

準備

下記サイトを参考にさせて頂きyolo v3の動作環境構築します。

YOLOv3をCPU環境でpython使ってリアルタイム画像認識を動かしてみた
 Macで物体検知アルゴリズムYOLO V3を動かす
 pytorch-yolo-v3

下記の通りcondaを使ってYolov3用の環境を構築して試しました。

$ conda create -n yolo_v3 python=3.6 pip

imgsフォルダに画像認識させたい.jpgファイルを格納してdetect.pyを実行します。

$ python detect.py --images imgs --det det

dog, bicycle.person...が認識されました
たまにdogがchair, teddy bearと認識されるのは一体なぜ..

Pexelsからダウンロードした動画にvideo_demo.pyを実行します。

$ python video_demo.py --video samplemovie.mp4

yolo v3 テスト pic.twitter.com/pDCnri5GhG
— st (@st17890027) July 8, 2019

$ python detect.py --images imgs --det det
RuntimeError: invalid argument Sizes of tensors must match except in dimension

batサイズの指定が誤っているようです。
下記サイトを参考に修正して動作させました。

RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0

You might add if output.size()[1] == prediction.size()[1]: condition, beforeoutput = torch.cat((output,prediction)), inside detect.py