【Ubuntu】YOLO v3

2021年6月4日

YOLO v3も試してみた。

こちらのサイトを参考にインストールします。実際は、venvで環境を作り、そこにインストールしました。

$ python3 -m venv yolov3
$ cd yolov3
$ source ./bin/activate
(yolov3)$

事前に、必要なパッケージ類をインストール。

(yolov3)$ pip3 install opencv-python opencv-contrib-python PyHamcrest torch torchvision pyyaml

これでも足りなければ、必要なものをその都度インストールします。

GithubからコードをCloneします。

(yolov3)$ git clone https://github.com/ultralytics/yolov3.git
(yolov3)$ cd yolov3

以下のコマンドでモデルがインストールされます。

(yolov3)$ ./weights/download_weights.sh

僕の環境では、以下の３つがダウンロードされました。

・yolov3-spp.pt
・yolov3-tiny.pt
・yolov3.pt

物体認識をしてみます。

(yolov3)$ python detect.py

data/imagesフォルダのbus.jpgとzidane.jpgの２つのファイルの内容を認識します。

結果は、runs/detect/exp* に保存されます。

bu.jpgとzidane.jpgが生成されました。

できた！

Webcamが接続されていれば、それを使って取得した画像でリアルタイムで物体検出できます。

python detect.py --source 0

PCが非力なので動きがカクカクしていますが、ちゃんと認識できます。すごい！

DiNOV