【深度学习】运行grad-cam(pytorch代码)遇到的一些问题

用原代码跑就没有问题,因为要提取自己数据集的判别性特征,而数据集加载必须使用pytorch自带的DataLoader。

这时候最后可视化热力图的时候图片明显不对,下面就是一点一点排除错误的过程。


 最后是直接将二值化后的mask找出坐标,裁剪出具有判别性的区域了!

你可能感兴趣的:(深度学习)