GitHub链接:
论文链接:
https://arxiv.org/pdf/1505.04597v1.pdf
复现:
1.从GitHub下载源码以后,配置U-net的环境。
matplotlib
numpy
Pillow
torch
torchvision
tensorboard
future
tqdm
2.配置完成后进行训练,但是我们没有数据集呀,这里我找到了一个医学细胞壁的数据集,只有30张灰度照片,各位老铁可以用这个进行训练~
百度网盘链接
链接:链接:https://pan.baidu.com/s/1EOuJzgs8UyPSCfJsxBDFuA
提取码:jt55
然后将其放在U-net master /data 下面 ,修改train.py文件的路径