参考文章:pytorch复现U-Net 及常见问题汇总(2021.11.14亲测可行)_奶盖芒果的博客-CSDN博客_pytorch 无法复现
1.安装过程
1.1 github代码
代码网址:https://github.com/milesial/Pytorch-UNet
1.2 环境配置
requirements:
matplotlib
numpy
Pillow
torch
torchvision
tqdm
wandb
requirements中未包含具体版本信息,笔者亲测torch11.3 + python3.6可用