# UNET-ZOO
including unet,unet++,attention-unet,r2unet,cenet,segnet ,fcn.
# ENVIRONMENT
window10(Ubuntu is OK)+pycharm+python3.6+pytorch1.3.1
## HOW TO RUN:
The only thing you should do is enter the dataset.py and correct the path of the datasets.
then run ~
example:
```
python main.py --action train&test --arch UNet --epoch 21 --batch_size 21
```
## RESULTS
after train and test,3 folders will be created,they are "result","saved_model","saved_predict".
### saved_model folder:
After training,the saved model is in this folder.
### result folder:
in result folder,there are the logs and the line chart of metrics.such as:
### saved_predict folder:
in this folder,there are the ouput predict of the saved model,such as:
### the datasets:
the Cell dataset(dsb2018)
link：https://pan.baidu.com/s/1BaVrzYdrSP78CwYaRzZr1w
keyword：5l54
the Liver dataset:
link：https://pan.baidu.com/s/1FljGCVzu7HPYpwAKvSVN4Q
keyword：5l88
the Cell dataset(isbi)
link：https://pan.baidu.com/s/1FkfnhU-RnYFZti62-f8AVA
keyword：14rz
the Lung dataset:
link：https://pan.baidu.com/s/1sLFRmtG2TOTEgUKniJf7AA
keyword：qdwo
the Corneal Nerve dataset:
link：https://pan.baidu.com/s/1T3-kS_FgYI6DeXv3n1I7bA
keyword：ih02
the Eye Vessels(DRIVE dataset)
link：https://pan.baidu.com/s/1UkMLmdbM61N8ecgnKlAsPg
keyword：f1ek
the Esophagus and Esophagus Cancer dataset from First Affiliated Hospital of Sun Yat-sen University
link：https://pan.baidu.com/s/10b5arIQjNpiggwdkgYNHXQ
keyword：hivm
UNet-zoo.zip （20个子文件） 
unet-zoo(Unet所有版本) 
segnet.py 6KB
lung.png 18KB UNet.py 5KB cell.png 55KB linechart.png 82KB main.py 15KB cenet.py 19KB dataset.py 16KB saved_predict init.py 1B saved_model init.py 1B metrics.py 6KB eye.png 66KB attention_unet.py 4KB README.md 2KB r2unet.py 3KB fcn.py 19KB unetpp.py 5KB plot.py 1KB channel_unet.py 5KB result init.py 1B
评论0