Bootstrap

U-net复现pytorch版本 以及制作自己的数据集并训练

GitHub链接:

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

论文链接:

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文件的路径

;