图片去噪训练模板
此模板用于训练图片去噪模型,下图为模板总览
- 首先就是4个数据源节点,分为两大类
- 训练图片,需要原始图片和带噪声的图片
- 测试图片,项目中也有原始图片和带噪声的图片两个节点,但是原始图片并不是必须的,仅作为对比用,可以不提供
- 训练图片,需要原始图片和带噪声的图片
- 其次就是去噪模型训练节点,用于训练去噪模型,以供导出做后面的预测用
- 输入桩
- 原始图片
- 噪声图片
- 权重文件:之前训练的模型,可以用于进一步训练(可不连)
- 原始图片
- 输出桩
- 输出模型
- Checkpoint:训练过程中的checkpoint
- Metric曲线文件夹:训练过程中产生的一些metric曲线图片
- 输出模型
- 参数
- 网络类型:选择需要使用的网络
- Batch Size:批大小
- 图片Patch大小
- Epoch:训练轮数
- 初始学习率
- 损失函数:mse,mae,l0
- steps:每一个epoch的step数
- 训练集比例:训练图片中,用于训练的图片占比(其余用于在每一个epoch结束后验证)
- 网络类型:选择需要使用的网络
- 输入桩
- 最后是预测节点,将测试图片进行去噪,对比效果
- 输入桩
- 原始图片(可不提供)
- 噪声图片
- 模型文件
- 原始图片(可不提供)
- 输出桩
- 预测结果
- 预测结果如下:
- 输入桩
- 从左到右分别为:噪声图片,去噪后图片,原图
- 当原始图片没有传入的时候,则只有噪声图片以及去噪后图片的对比
- 当模型文件达到预期之后,可以点击模型训练节点中的导出模型按钮,将训练得到的模型导出为一个模型节点,给其他项目使用。新导出的模型节点可以在 我的组件列表中找到