一、開發(fā)環(huán)境
- Windows 10
- PyCharm 2021.3.2
- Python 3.7
- PyTorch 1.7.0
二、制作交通標(biāo)志數(shù)據(jù)集,如下圖
三、配置好數(shù)據(jù)集的地址,然后開始訓(xùn)練
python train.py --data traffic_data.yaml --cfg traffic_yolov5s.yaml --weights pretrained/yolov5s.pt --epoch 100 --batch-size 4
四、訓(xùn)練完成后進(jìn)行推理預(yù)測,效果如圖
五、完整源碼、數(shù)據(jù)集和模型文件下載文章來源:http://www.zghlxwxcb.cn/news/detail-702610.html
鏈接:https://pan.baidu.com/s/111wLXWLckTfrHIEGZyXeqA?pwd=w5q6
提取碼:w5q6文章來源地址http://www.zghlxwxcb.cn/news/detail-702610.html
到了這里,關(guān)于基于PyTorch的交通標(biāo)志目標(biāo)檢測系統(tǒng)的文章就介紹完了。如果您還想了解更多內(nèi)容,請?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!