基于PyTorch的交通标志目标检测系统

一、开发环境

• Windows 10
• PyCharm 2021.3.2
• Python 3.7
• PyTorch 1.7.0

二、制作交通标志数据集，如下图

三、配置好数据集的地址，然后开始训练

python train.py --data traffic_data.yaml --cfg traffic_yolov5s.yaml --weights pretrained/yolov5s.pt --epoch 100 --batch-size 4

四、训练完成后进行推理预测，效果如图

五、完整源码、数据集和模型文件下载

链接：https://pan.baidu.com/s/111wLXWLckTfrHIEGZyXeqA?pwd=w5q6
提取码：w5q6