用原代码跑就没有问题,因为要提取自己数据集的判别性特征,而数据集加载必须使用pytorch自带的DataLoader。
这时候最后可视化热力图的时候图片明显不对,下面就是一点一点排除错误的过程。
最后是直接将二值化后的mask找出坐标,裁剪出具有判别性的区域了!
用原代码跑就没有问题,因为要提取自己数据集的判别性特征,而数据集加载必须使用pytorch自带的DataLoader。
这时候最后可视化热力图的时候图片明显不对,下面就是一点一点排除错误的过程。
最后是直接将二值化后的mask找出坐标,裁剪出具有判别性的区域了!