
UNET-ZOO涵盖了unet、unet++、attribute-unet、r2unet、cenet、segnet和fcn等模型。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
UNET-ZOO 涵盖了 unet、unet++、attribute-unet、r2unet、cenet、segnet 和 fcn 等多种模型。该环境支持 Windows 10 以及 Ubuntu,并需要配合 PyCharm 和 Python 3.6 (以及 PyTorch 1.3.1) 进行运行。 具体操作步骤为:只需输入 dataset.py 文件,并相应地调整数据集的路径即可。 接下来,执行命令 `python main.py --action train&test --arch UNet --epoch 21 --batch_size 21`。 在训练和测试完成后,程序将自动生成三个独立的文件夹,包括“结果”、“saved_model”和“saved_predict”这三个目录。 “saved_model”文件夹中存储了训练过程中保存下来的模型文件。 “结果”文件夹内则包含了训练指标的日志记录以及各种折线图等可视化数据。“saved_predict”文件夹则用于存放预测结果。
全部评论 (0)
还没有任何评论哟~


