UNET-ZOO
包括unet,unet ++,attribute-unet,r2unet,cenet,segnet,fcn。
环境
window10(Ubuntu可以)+ pycharm + python3.6 + pytorch1.3.1
如何运行:
您唯一要做的就是输入dataset.py并更正数据集的路径。 然后运行〜示例:
python main.py --action train&test --arch UNet --epoch 21 --batch_size 21
结果
经过训练和测试后,将创建3个文件夹,分别是“结果”,“ saved_model”,“ saved_predict”。
saved_model文件夹:
训练后,保存的模型在此文件夹中。
结果文件夹:
在结果文件夹中,有指标的日志和折线图,例如:
saved_predict文件夹:
在此文件夹中,
1