pytorch_grad_cam可视化类别激活热力图报错：axis 2 is out of bounds for array of dimension 0

Question

我在利用pytorch_grad_cam对mobileone模型进行可视化特征时报了下面的错误，但改用resnet50的模型没问题，请问有人遇到过吗？

Answer 1

分类任务使用Pytorch实现Grad-CAM绘制热力图_pytorch 热力图-CSDN博客 文章浏览阅读2.1k次，点赞4次，收藏23次。给大家打个比方，下图是我输出模型结构的一小块，此时如果我设定target_layers=[model.patch_embed_a]，那它就相当于被我设置成一个元组，但这样是无法应用在这个代码上的，我们需要继续细化到某一层，比如改成target_layers=[model.patch_embed_a.norm]这样就可以成功绘制热力图。对于深度学习网络，在我们指定数据集类别的情况下，Grad-CAM能够绘制出相应的热力图，让我们能够非常直观的看出网络关注的主要区域与特征是什么。_pytorch 热力图 https://blog.csdn.net/Linchpin_/article/details/129796835