概述
YOLOv5-5.0版本的四個預(yù)訓(xùn)練權(quán)重輸入的圖片尺寸固定為640x640。
但是在YOLOv5-6.1版本，多了幾個例如yolov5l6.pt末尾帶6的預(yù)訓(xùn)練權(quán)重，這幾個權(quán)重是在更大分辨率1280x1280進行訓(xùn)練的。
因此我想在自己的電腦上測試一下相同的數(shù)據(jù)集從640x640到1280x1280效果會有多少提升。
原本是想在VisDrone數(shù)據(jù)集上進行對比測試，然后當(dāng)換用更大分辨率之后，即使batch_size減為1，依然面臨顯存不足的問題。
于是我使用了自己的數(shù)據(jù)集，采用yolov5l模型，跑了100個epoch，mAP效果如下表所示：
| 算法 | mAP@.5 | mAP@.5:.95 |
|---|---|---|
| yolov5(640) | 95.6% | 67.6% |
| yolov5(1280) | 98.7% | 75.7% |
注：自己的數(shù)據(jù)集樣本量不多，并且目標較稀疏，因此mAP比較高。文章來源：http://www.zghlxwxcb.cn/news/detail-457426.html
從表中可以看出，換用更大尺寸的分辨率之后，mAP指標還是有明顯提升的。
下面是Result.png
yolov5(640)：
yolov5(1280)：
如圖所示，可以看到y(tǒng)olov5(1280)的收斂速度更快一些。文章來源地址http://www.zghlxwxcb.cn/news/detail-457426.html
到了這里，關(guān)于【目標檢測】YOLOv5：640與1280分辨率效果對比的文章就介紹完了。如果您還想了解更多內(nèi)容，請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章，希望大家以后多多支持TOY模板網(wǎng)！
