環境

種別	名称	バージョン	備考
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フィールドを参照