Day21
整理代码的时候发现cbam一直用的hic里的CBAMBottleneck,改成了ultralytics提供的CBAM,map竟然涨了不少,太奇怪了
新加了SEAttention,合并一下以前的结果
VisDrone2019,200epochs
Model | Parameters(M) | mAP@50:95(%) | mAP50(%) |
---|---|---|---|
yolov5s | 9.15 | 21.12 | 36.06 |
yolov5s-cbam | 9.42 | 22.54 | 38.00 |
yolov5s-se | 9.19 | 22.57 | 38.04 |
居然有涨点啊,该试试yolov8了,不过要测试不同的CBAM添加的位置
CBAM添加位置有如下几个情况:
- SPFF前一层
- Detect前一层
- 每个检测头后一层
- backbone里每个C2f/C3后一层
在yolov5中加入了transform layer试试
本文由作者按照 CC BY 4.0 进行授权