Yummy1989的博客

原创 MMSegmentation多个输入使用Grad-Cam可视化方法

注：我的模型仅支持正方形输入，例如512，所以我在处理前，把图像手动resize到512*512，然后在脚本的最后，resize回原大小，例如640*480，这样注意力就是在原图画的啦。这么做的原因是：送往模型肯定是要修改为双模态，因为我们的模型是双模态的，但是gradcam在画图的时候进需要一个图像进行可视化。grad-cam只支持单输入，但是对于多输入模型，例如：红外和可见光输入模型，如何使用grad-cam可视化？数据预处理部分：同上，并把红外和可见光图像按照维度拼接起来。至此，应该修改完毕了。