| 種別 | 名称 | バージョン | 備考 |
|---|---|---|---|
| OS | Windows | 10 | |
| GPU | GeForce RTX 3090 | ||
| CUDA | 11.3 | ||
| lib | pytorch | 1.12.1+cu113 |
python -m venv .env
./.env/Scripts/Activate.ps1
pip install -r requreirements.txt使用したモデルはこちら
他のモデルで実行したい場合はこちらの表のlinkから選ぶ
python set_border_rect.py実行後
left_late = 0
top_late = 0.4
right_late = 1
bottom_late = 1の値を0-1の実数で境界にしたい位置まで設定する
python main.py --weights ./yolov5x6.pt --source 0認識させるクラスを絞る場合は--classes 0 5 7のようなオプションをつける
オプション後半の数字は./data/coco128.yamlファイルのnamesフィールドを参照