【Ubuntu】yolov5

こちらを参考にyolov5を使ってみます。

1. インストール

まず、クローンします。

git clone https://github.com/ultralytics/yolov5

環境を作ります。

python3 -m venv yolov5
cd yolov5
source ./bin/activate

PyTorchをインストールします。 （不要でした。次のrequirementsから最新版をインストールします。）

pip install torch===1.5.1 torchvision===0.6.1 -f https://download.pytorch.org/whl/torch_stable.html

requirements.txtからもろもろインストールします。

pip install -U -r requirements.txt

2. つかってみる

python detect.py --source data/images/bus.jpg

実行すると、runs/detect/exp*に結果のファイルが出力されます。