Day22 - CBAM在yolov5s上不同添加位置在VisDrone上的对比实验
对CBAM不同添加位置的结果
- yolov5s-cbam0: SPPF前一层
- yolov5s-cbam1: Detect前一层
- yolov5s-cbam2: 每个检测头后一层
- yolov5s-cbam3: backbone里每个C2f/C3后一层
VisDrone2019,200epochs
Model | Parameters(M) | mAP@50:95(%) | mAP50(%) |
---|---|---|---|
yolov5s | 9.15 | 22.69 | 38.05 |
yolov5s-cbam0 | 9.42 | 22.54 | 38.00 |
yolov5s-cbam1 | 9.42 | 22.57 | 38.04 |
yolov5s-cbam2 | 9.50 | 22.57 | 38.04 |
yolov5s-cbam3 | 9.50 | 22.53 | 38.31 |
cbam除了加参数量屁用没有啊,当凑工作量算了,最后留下cbam1
本文由作者按照 CC BY 4.0 进行授权