including unet,unet++,attention-unet,r2unet,cenet,segnet ,fcn.
window10(Ubuntu is OK)+pycharm+python3.6+pytorch1.3.1
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
after train and test,3 folders will be created,they are "result","saved_model","saved_predict".
After training,the saved model is in this folder.
in result folder,there are the logs and the line chart of metrics.such as:
in this folder,there are the ouput predict of the saved model,such as:
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