YOLO·v5 Grad CAM运行出错keyerror:cv3解决方法

最近在搞YOLOv5的热力图，但是网上的代码，用原来的模型没报错，但是用上自己加上注意力训练出来的模型报错了，cv3

解决方法

 原先的v5有23层，加上注意力机制之后网络层数发生了变化，所以这也需要改成对应的层数

YOLOv5 Grad CAM的代码，看这位博主的，其实github上也有都是一样的

【YOLOV5从开发到部署实战】十一、GradCAM可视化显示YOLOv5热力图_落难Coder的博客-CSDN博客_yolov5热力图

这个程序只能在锚框内生成热力图，如果想要那种全图的效果，可以去买白勇老师的课，那里面的代码是可以做到全图的，我试过，花钱买的，就不给大家展示了。

说实话，有些论文里面放了这种图，只不过也是为了证明注意力机制的效果，但是这种图也比较占篇幅，有没有都行，顶多加上这种图可能论文好看一点