从一次诡异的漏检说起
上周在产线部署YOLOv5时遇到个怪事:同一个型号的电路板,白天检测正常,晚上总有固定几个焊点被漏检。光照条件模拟过了,数据增强也加了,问题依旧。最后把模型中间层特征图拉出来看,才发现某个通道对高光反射特别敏感——模型压根没学会看焊点形状,而是依赖反光纹理做判断。这事儿让我彻底明白:黑盒模型要上线,不打开看看心里真没底。
为什么YOLO需要可解释性?
工业场景里,模型光有高mAP不够。客户会问:“为什么这里判成缺陷?”“漏检的根因是什么?” 传统的调参-测试循环回答不了这些问题。XAI(可解释AI)不是学术玩具,而是工程必需品。尤其YOLO这种one-stage检测器,决策过程更隐蔽,可视化能帮我们看清三件事:模型在看哪里、用什么特征判的、置信度怎么来的。
可视化YOLO决策的实操路线
1. 特征图可视化:看模型“眼里”的世界
直接上代码,这里用YOLOv5举例,但思路通用:
import torch
import cv2
import numpy as np
from






