跑语义分割网络Unet出现的问题

GitHub：GitHub - milesial/Pytorch-UNet: PyTorch implementation of the U-Net for image semantic segmentation with high quality images数据集：

源于U-net复现pytorch版本 以及制作自己的数据集并训练_pytorch用u-net训练自己的数据集_奶盖芒果的博客-CSDN博客

另外基于pytorch编写Unet进行细胞结构分割_细胞分割pytorch-CSDN博客这篇用的也是细胞壁数据集

百度网盘链接

链接：链接：https://pan.baidu.com/s/1EOuJzgs8UyPSCfJsxBDFuA 
提取码：jt55 

问题：

1.修改train,py和predict,py中的channel，如果是RGB图像，channel=3，如果是灰度图，channel=1；

2.outofmemory

虚拟内存加大到20480

3.unet oserror  loading cudnn_adv_train64_8.dll  or one of its dependencies

修改batchsize没用，

在train.py文件里修改num_worker为2，如果还是不行，改成0 

OSError: [WinError 1455] 页面文件太小,无法完成操作_winerror 1455 页面文件太小,无法完成操作-CSDN博客

训练过程：

训练结果：

Dice值接近1说明训练结果理想

预测输入图&输出图