跳到主要内容

图片去噪训练模板

此模板用于训练图片去噪模型,下图为模板总览 image.png

  • 首先就是4个数据源节点,分为两大类
    • 训练图片,需要原始图片和带噪声的图片
    • 测试图片,项目中也有原始图片和带噪声的图片两个节点,但是原始图片并不是必须的,仅作为对比用,可以不提供
  • 其次就是去噪模型训练节点,用于训练去噪模型,以供导出做后面的预测用
    • 输入桩
      • 原始图片
      • 噪声图片
      • 权重文件:之前训练的模型,可以用于进一步训练(可不连)
    • 输出桩
      • 输出模型
      • Checkpoint:训练过程中的checkpoint
      • Metric曲线文件夹:训练过程中产生的一些metric曲线图片
    • 参数
      • 网络类型:选择需要使用的网络
      • Batch Size:批大小
      • 图片Patch大小
      • Epoch:训练轮数
      • 初始学习率
      • 损失函数:mse,mae,l0
      • steps:每一个epoch的step数
      • 训练集比例:训练图片中,用于训练的图片占比(其余用于在每一个epoch结束后验证)
  • 最后是预测节点,将测试图片进行去噪,对比效果
    • 输入桩
      • 原始图片(可不提供)
      • 噪声图片
      • 模型文件
    • 输出桩
      • 预测结果
      • 预测结果如下:

image.png

  - 从左到右分别为:噪声图片,去噪后图片,原图
- 当原始图片没有传入的时候,则只有噪声图片以及去噪后图片的对比
  • 当模型文件达到预期之后,可以点击模型训练节点中的导出模型按钮,将训练得到的模型导出为一个模型节点,给其他项目使用。新导出的模型节点可以在 我的组件列表中找到

image.png image.png image.png