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给的官方接口,干净利落。