unet-pytorch
unet-pytorch copied to clipboard
同一张图片在两个不同的模型预测下有时能成功预测出结果,有时输出一张灰色图片。使用的两次预测模型就是在Unet基础上稍加修改了一下,加入了不同的注意力机制
预测结果像这样:

你修改后没有训练就预测了?
训练了的,预测的时候载入的都是训练得到的不同的权重文件
你要不描述清楚一些吧,哪一些得到什么结果,用的什么模型
预测结果像这样:
请问你在unet中加入的注意力机制,是加在哪了