训练模型

完成标注后,即可开始训练模型。本文介绍如何在“训练”参数栏配置训练参数、训练模型、查看训练信息。

配置训练参数

开始训练前,应先为深度学习模型配置训练参数。单击 参数配置 打开“训练参数配置”窗口,可配置数据增强参数、训练参数与模型微调参数(仅开发者模式)。

为深度学习模型调整最优参数需要进行多次尝试,首次训练建议使用 Mech-DLK 提供的默认参数值,可适用于大多数应用场景。
training parameters configuration

数据增强

训练模型的数据需要尽量包含实际可能出现的所有情况,若现场不具备相应的采集条件,可以针对缺失的情况在训练前调整数据增强参数,在原有数据的基础上丰富训练数据。但必须确保倍增的图像数据与现场实际情况一致,如果现场不会出现旋转的情况,那么不需要调整这一参数,否则可能会影响模型效果。

鼠标悬浮在 train ask icon 可查看各参数的调节效果。
  • 亮度

    图像的明暗程度。现场光照变化较大时,可以通过调整 亮度 范围倍增不同光照状态下的图像数据。

  • 对比度

    图像灰度反差的大小。目标物体和背景差异较小时,适当调整 对比度 范围,可以使目标物体特征更明显。

  • 平移

    将图像所有的像素坐标分别加上指定的水平偏移量和垂直偏移量。现场物体(料筐、托盘等)位置移动范围较大时,可以通过调整 平移 参数,随机倍增横向、纵向平移的数据。

  • 旋转

    指图像以某一点为中心旋转一定的角度,形成一张新的图像。一般情况下保持默认参数设定即可满足需求。物体摆放角度变化较大时,通过调整 旋转 参数,可以随机倍增多种角度的图像数据。

  • 缩放

    将图像按一定比例缩小或放大。物体摆放高度差异较大时,通过调整 缩放 参数 ,可以随机倍增不同缩放程度图像数据。

  • 水平翻转

    图像左右翻转 180°。若待目标物体存在左右对称的情况,可以开启 水平翻转

    developer mode flip vertically
  • 垂直翻转

    图像上下翻转 180°。若目标物体存在上下对称的情况,可以开启 垂直翻转

    developer mode flip horizontally

训练参数

  • 模型类型

    “缺陷分割”模块

    高速度

    适用于对训练速度要求较高的场景

    高精度

    适用于对精度要求较高的场景,该模式会导致训练速度变慢

  • 输入图像尺寸

    训练时输入神经网络的图像宽高(单位:像素)。建议使用默认设定;若图像中的物体或标注缺陷区域较小时,需要适当调大 输入图像尺寸 。图像尺寸越大,模型精度越高,但是训练速度越慢。

  • 批量大小

    神经网络训练一次选取的样本数量。建议使用默认设定;若需要提高训练速度时,可以适当调大 批量大小 参数,参数设置太大会导致显存占用变高。

  • 评估间隔轮次

    模型训练时,每评估一次间隔的轮次数量。建议使用默认设定,调大 间隔轮次 参数可以提升训练速度 ,参数值越大训练越快;参数值越小训练越慢,但有助于挑选最优模型。

  • 总轮次

    模型训练的总轮次。建议使用默认设定,若待识别图像特征较复杂时,需要适当增加训练轮次,提升模型效果,但会导致训练时间变长。

    训练轮次并非越大越好,总轮次设置较大时,模型准确率稳定后还会持续训练,会导致训练时间变长,且有过拟合的风险。
  • 学习率

    学习率决定神经网络训练过程中每一次迭代优化的步长。建议使用默认设定,图表损失曲线收敛较慢时,可以适当调大学习率;若图表精度波动较大,可以适当调小学习率

  • GPU ID

    用户训练设备的显卡信息。若用户设备有多块 GPU ,可以在指定的 GPU 上训练。

  • 训练时最大目标个数

    该选项用于限定能识别到的最大目标个数,默认不勾选,勾选并设定数值后一般能加快推理速度。一定情况下,设定此参数有助于减少GPU显存使用。

模型微调(“开发者模式”下可见)

在模型上线一段时间后,可使用该功能对无法覆盖更多使用场景的模型进行迭代。

详细操作说明请参考迭代模型

开始训练

确认参数修改后,在训练参数栏下方,单击训练开始训练模型。

训练中心

在训练中心可查看模型训练进度与显存占用情况。

如果已有训练任务正在进行中,其他加入训练的工程会进入等待训练队列。

training center
  • 单击 train center stop 终止正在训练的任务。

  • 单击 train center delete 将当前任务从等待队列中删除。

  • 单击 train center change 将当前任务调整到等待训练队列的顶部。

查看训练信息

  • “训练”参数栏中会实时展示训练信息和训练任务中达到的最优结果。

  • 在训练参数栏,单击显示图表按钮,可将当前图像切换为训练参数图表。该图表实时展示了随训练轮次增长,训练精度和损失参数的变化。再次单击显示图表按钮可切换回图像界面。

view training info

我们重视您的隐私

我们使用 cookie 为您在我们的网站上提供最佳体验。继续使用该网站即表示您同意使用 cookie。如果您拒绝,将使用一个单独的 cookie 来确保您在访问本网站时不会被跟踪或记住。