训练模型

使用Mech-DLK训练模型

参考 快速入门

可调整的训练参数

多数情况使用默认参数进行训练即可满足需求,以下参数仅在有特殊需求时按需调整。

../../../_images/training_parameter.png
亮度范围
现场光照变化较大,通过补光和遮光也无法弥补时,采集的训练数据光照差异也会比较大。这种数据用于训练需要适当调大亮度范围,差异较小则保持默认参数。
对比度范围
物体和背景的差异不明显时 为了模型更好地学习物体特征,可适当调大范围,通常与亮度范围配合使用。由于此种情况极少发生,一般无需调整。
平移范围
若现场物料背景(料筐、托盘或其他)位置移动范围较大,需调大平移范围,反之则保持默认参数。
旋转范围
当物体在固定位置需区分不同方向时,需将范围调为0-0,避免因旋转而影响模型学习方向特征,其他情况时保持默认参数。
缩放范围
当物体 位置高度差异大 或者同等高度时物体间 体积差异较大 时需调大缩放范围,其他情况时保持默认参数。
翻转
当物体在固定位置需区分不同方向时,需关闭翻转选项,避免因翻转而影响模型学习方向特征,通常与旋转范围配合使用,其他情况时保持默认参数。
通道随机混合
依靠颜色分类且形状差异不大 时,需关掉此选项,避免因颜色混合变化而影响模型对物体颜色特征的学习,其他情况时保持默认参数。
总训练轮次
不分类、物体特征简单时,可设置在600epoch内取最优模型;物体姿态复杂,种类多或其他特征复杂时可设置在1000epoch内取最优模型。
学习率
一般情况下无需调整,当准确率数值较低(<0.8)或突然下降时,可调整为当前学习率数值的十分之一。

注意

如果调整为实际没有的范围,反而会影响模型效果;例如当光照稳定时,过多调整亮度范围,模型效果则不如默认参数值下稳定。