训练分割小工件的高精度模型
本文以3D视觉引导曲轴上料为例,介绍在视野范围大、工件尺寸小的场景下,如何通过级联算法模块训练一个高精度的深度学习模型。
训练流程
曲轴自动化上料需要引导机器人从托盘中识别并抓取散乱放置的曲轴,可通过级联模块实现以下三个主要步骤:
-
使用目标检测模块定位装有曲轴的托盘位置。
-
使用实例分割模块分割出托盘上放置的曲轴。
-
使用目标检测模块定位曲轴大端,便于抓取。
需要根据各个模块的识别目标有针对性地进行数据采集及标注,才能训练出更精确的模型。
数据采集
调整相机的2D参数
-
高分辨率模式:推荐使用高分辨率相机采集图像进行训练。本案例中,LSR L-V4相机使用1200万像素的高分辨率模式。
-
曝光模式:由于环境光差异较大,并且背景托盘颜色多样,建议将RGB相机的曝光模式设置为自动曝光,并将自动曝光感兴趣区域设置为托盘。
-
灰度值:曲轴为高反光工件,建议调整灰度值参数,确保在深色和浅色托盘背景下,工件图像均不过曝。