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

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

解决方法

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

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

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

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

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

你可能感兴趣的:(深度学习,人工智能)