深度学习笔记-实验过程/代码结构

代码结构

dataset.py

1. 从文件夹中读取数据
2. 按照文件尺寸大小进行中心裁剪【FCN中的思路】
3. 标签编码
4. numpy转tensor
5. 实例化数据类

train.py

1. 读取数据
2. 加载模型
3. 配置相关参数
4. 训练并输出结果（指标计算）
5. 验证并输出结果（指标计算）
6. 保存最好的权重

test.py

1. 读取数据
2. 加载已经保存的最好权重
3. 测试并输出结果（指标计算）

predict.py

1. 读取数据
2. 加载已保存的最好权重
3. 得到预测图并上色
4. 输出彩色预测图片