- 实现了行人出入分别计数,沿着图像上下方向检测。
- 检测类别:行人、自行车、小汽车、摩托车、公交车、卡车。
-支持yolov5s.pt yolov5x.pt yolov5m.pt yolov5l.pt模型
代码、模型和demo视频下载地址:下载地址
环境配置:
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 -f /whl/torch_stable.html
- python 3.6+,pip 20+- pytorch- pip install -r requirements.txt
运行命令:
python main.py
部分源码解读:
import numpy as npimport trackerfrom detector import Detectorimport cv2if __name__ == \__main__