097、YOLOv11改进-论文中详述改进点的动机分析与可视化对比——即插即用模块的注意力热图与特征可视化实战

097、YOLOv11改进-论文中详述改进点的动机分析与可视化对比——即插即用模块的注意力热图与特征可视化实战

上周调一个CBAM改进的YOLOv11，mAP涨了0.3个点，审稿人回了一句“动机分析不够直观，建议补充可视化证据”。我当时就想骂人——明明涨点了，非要我画图证明为什么涨点。后来想通了，论文里没有可视化对比，审稿人根本不知道你改的那个模块到底在干什么。今天这篇笔记，我就把怎么用代码把注意力热图和特征图可视化做出来，顺便聊聊怎么把这些图写成论文里能说服人的动机分析。

从一次翻车现场说起

事情是这样的。我在YOLOv11的Neck里塞了一个改进的SimAM模块，训练完发现小目标检测确实好了，但审稿人问“你的模块为什么比SE好？凭什么说它更关注空间位置？”我当时拿不出证据，只能硬着头皮补实验。后来我写了个可视化脚本，把特征图抽出来，用热力图叠在原图上，对比SE和SimAM的注意力分布——结果一目了然：SE把注意力均匀撒在整个图上，SimAM聚焦在目标边缘和纹理区域。这个对比图直接成了论文Figure 3，审稿人再没废话。

所以，可视化不是花架子，是你改进模块的“呈堂证供”。

怎么从YOLOv11里把特征图捞出来

YOLOv11的模型结构用nn.Module堆起来的，想拿到中间层的输出，最直接的办法是注册hook。别用那种改模型forward函数的方法，太脏了，hook是PyTorch给的官方接口，干净利落。