Day19
yolov5s为基准,200epochs,在VisDrone2019上训练结果
Model | Parameters(M) | mAP@50:95(%) | mAP50(%) |
---|---|---|---|
yolov5s | 9.15 | 22.84 | 38.22 |
yolov5s-p2 | 8.82 | 25.72 | 42.66 |
yolov5s-p234 | 6.27 | 25.71 | 42.62 |
yolov5s-cbam | 13.19 | 21.82 | 37.47 |
yolov5s-cbam-p234 | 7.24 | 25.17 | 42.77 |
对于yolov5s,加了cbam精度反而下降了,过拟合了吗
虽然yolov5s-cbam-p234的map最高,但是也就0.1个点,可以看出cbam的贡献微乎其微
后面的工作就着重在修改backbone和cbam调参上吧
看了一个yoloair项目,他把cbam放在head里,于是我也改了一下
yolov5s为基准,100epochs,在VisDrone2019上训练结果
Model | Parameters(M) | mAP@50:95(%) | mAP50(%) |
---|---|---|---|
yolov5s | 9.15 | 21.12 | 36.06 |
yolov5s-cbam | 9.42 | 21.03 | 35.75 |
效果依然不乐观,用yolov8m再跑一遍
效果依然不行,放弃了,找了acmix的改进代码,尝试跑一下
本文由作者按照 CC BY 4.0 进行授权