首先数据预处理确认是正确的,在unet模型上已经运行成功,当数据集在segnet模型上时,会出现如下错误:
RuntimeError: reduce failed to synchronize: device-side assert triggered
百度说是标签的index出错,应该是从0开始的,但是我在读数据时,它就是自动从零开始,所以这个原因不可行。
后来检查代码发现模型的输出没有用激活函数
torch.nn.Sigmoid()(x11d)
在模型的最后添加这个就成功解决啦!