实现了 出/入 分别计数。
显示检测类别。
默认是 南/北 方向检测,若要检测不同位置和方向,可在 main.py 文件第13行和21行,修改2个polygon的点。
默认检测类别:行人、自行车、小汽车、摩托车、公交车、卡车。
检测类别可在 detector.py 文件第60行修改。
运行环境
python 3.9.10,pip 22.0.3+
pytorch 1.10.2+
pip3 install -r requirements.txt
如何运行
确保正确安装 python 和 CUDA
D:\> python -V
D:\> nvidia-smi
D:\> nvcc -V