- 博客(2)
- 收藏
- 关注
RSS订阅原创 分类任务使用Pytorch实现Grad-CAM绘制热力图
给大家打个比方，下图是我输出模型结构的一小块，此时如果我设定target_layers=[model.patch_embed_a]，那它就相当于被我设置成一个元组，但这样是无法应用在这个代码上的，我们需要继续细化到某一层，比如改成target_layers=[model.patch_embed_a.norm]这样就可以成功绘制热力图。对于深度学习网络，在我们指定数据集类别的情况下，Grad-CAM能够绘制出相应的热力图，让我们能够非常直观的看出网络关注的主要区域与特征是什么。
2023-03-27 15:54:50
4550
2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人