文章

Day21

整理代码的时候发现cbam一直用的hic里的CBAMBottleneck,改成了ultralytics提供的CBAM,map竟然涨了不少,太奇怪了

新加了SEAttention,合并一下以前的结果

VisDrone2019,200epochs

ModelParameters(M)mAP@50:95(%)mAP50(%)
yolov5s9.1521.1236.06
yolov5s-cbam9.4222.5438.00
yolov5s-se9.1922.5738.04

居然有涨点啊,该试试yolov8了,不过要测试不同的CBAM添加的位置

CBAM添加位置有如下几个情况:

  • SPFF前一层
  • Detect前一层
  • 每个检测头后一层
  • backbone里每个C2f/C3后一层

在yolov5中加入了transform layer试试

本文由作者按照 CC BY 4.0 进行授权