yolov8热力图可视化与NoneType‘ object has no attribute ‘shape‘报错解决

目录

一、参考

二、热力图文件参数设置

1、第一个：权重

2、第二个：配置文件

如果你用的预训练模型怎么办？

三、如标题-报错解决

参考：

---

一、参考

参考这位博主大大就好啦，写的很好

YOLOV8-gradcam 热力图可视化 即插即用 不需要对源码做任何修改!_yolov8热力图-CSDN博客

二、热力图文件参数设置

关于参数设置，我有点想说：

1、第一个：权重

可以是你自己训练好的模型best.pt，也可以是预训练模型yoloc8n之类的

2、第二个：配置文件

不是关于你数据的那个文件，是里面有head，backbone的那个文件，也就是内容是下面这样的文件：

这个一定要完全对应与你当初训练时用的yaml文件。

如果你用的预训练模型怎么办？

这里假设你用的是yolov8s模型，那么你需要找到yolo.yaml这个配置文件，因为v8五个预训练模型都对应的这个。路径在

里面的v8模块下就有，然后给他改个名，改成yolov8s.yaml，然后把路径连同文件名输入第二个位置下

(因为如果不改，我们就不知道你训练的是哪个预训练模型，那就会默认用n模型的yaml配置参数，就会报错)

如下

三、如标题-报错解决

我有个报错

报错：AttributeError: 'NoneType' object has no attribute 'shape'

本质是你的图片，没有被读到。所以你要仔细检查你的路径是否正确，图片名称是否一致，路径不要有中文。以上

1、图片不存在（检查图片名是否正确）；
2.读取的图片内容和默认读取时参数匹配不匹配。（默认读取的是3通道的彩色图）例如读取到的图片是灰度图，就会返回None；
3、所读图片所在路径中包含中文（笔者就是犯了这样的错误，后来将保存代码文件的文件夹改成了英文）；

参考：

AttributeError: 'NoneType' object has no attribute 'shape'解决方案-CSDN博客（感谢大大）

我的报错原因是因为我后来把一个文件改名了，但是路径里面忘改了，所以没读到也是必然的了

改了正常路径后就没错了