pytorch复现U-Net 及问题

U-NET安装与训练问题实录
pytorch复现U-Net 及常见问题汇总（2021.11.14亲测可行）

以上给我的帮助很大。

遇到的问题
1. OSError: [WinError 1455] 页面文件太小，无法完成操作：
方法一：

这个直接说解决方法，在train.py文件里修改
num_worker为2，如果还是不行，改成0

方法二：修改高级系统设置中的性能选项

2.INFO: Unique mask values: [0, 1] ERROR: Retry attempt failed: Traceback (most recent call last): File “D:\Anaconda\Anaconda\envs\WH\lib\site-packages\urllib3\connection.py”, line 198, in _new_conn sock = connection.create_connection…
这种我搜了很多资料都不行，发现别人下载了wandb，我去下载了wandb，才发现问题。
关于wandb的安装使用：
wandb在pycharm中安装与配置经验分享
当然，wandb需要翻墙，不然根本连接不上。

安装完还是有问题：

这是wandb在和官网链接不上，直接wandb离线,在指令行输入：
wandb offline
之后在运行train.py就可以了。

训练完train.py后该是要修改predict.py的参数，参考下面文章的操作。
UNet复现过程中遇到的问题以及解决

train.py操作成功：（epoch是5轮，图片上是第1轮）

predict.py操作成功图片：