本项目是本人授课使用,请仅做个人学习、研究使用。
- PC端运行:Windows10或11(最好有GPU)
- MACOS(可能比较卡)
- Pytorch
- YOLOv5
- opencv
-
安装第三方包:
pip install -r requirements.txt
-
下载相关文件:下载地址
- 下载训练好的权重文件
highway_m_300.pt
和highway_reid.pt
放到weights
目录下; - 下载字体
MSYH.ttc
至fonts
目录下; - 下载媒体文件
highway.mp4
至videos
目录下(该视频是本人无人机拍摄,禁止他用)
- 下载训练好的权重文件
-
运行
python demo.py
即可。