一、开发环境
- 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









![[国产MCU]-W801开发实例-MQTT客户端通信](https://img-blog.csdnimg.cn/f6159728c6d74fc8bb888e440eddc698.png#pic_center)





![[github-100天机器学习]day4+5+6 Logistic regression](https://img-blog.csdnimg.cn/d4c502b7452343e9b4da0dfbd52be367.png)



