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