yolov5 行人 车辆 跟踪 检测 计数

实现了 出/入 分别计数。

显示检测类别。

默认是 南/北 方向检测，若要检测不同位置和方向，可在 main.py 文件第13行和21行，修改2个polygon的点。

默认检测类别：行人、自行车、小汽车、摩托车、公交车、卡车。检测类别可在 detector.py 文件第60行修改。

在 main.py 文件中第66行，设置要检测的视频文件路径，默认为 './video/test.mp4'

capture = cv2.VideoCapture('./video/test.mp4')

2.运行程序

python main.py