跳转至

五、 额外部分:数据增强

功能说明

  • 目前该功能并未集成在 「ATSVisionOps」 当中,属于另外单独的程序功能,但不排除后期我会将其功能集成在 「ATSVisionOps」 当中。

通过数据增强功能,您可以基于现有样本生成更多样化的训练数据,从而提升模型在工业现场的抗干扰能力和检测精度。


1. 开启功能

打开 训练数据打包工具

sjzq1.png

在主界面双击进入 「目标检测数据集」 应用程序。


2. 增强算子说明

sjzq2.png

请根据您的产品特征(如圆形、方形、表面纹理等)选择合适的镜像或变换方式:

增强模式 核心逻辑 适用场景
横向镜像 以 Y 轴为中轴线进行左右翻转。 适用于框标图,如区分产品的内外侧。
纵向镜像 以 X 轴为中轴线进行上下翻转。 适用于框标图,增加位置鲁棒性。
旋转一周 以图像圆心为原点进行旋转,角度可自由调整。 推荐用于点标图,常用于圆形表面缺陷检测。
色度/对比度 在限定区间内随机调整图像色彩与明暗。 默认设置即可,用于模拟产线光照变化。

3. 参数配置

在界面右侧或下方配置以下关键参数:

  • 图数:指在设定的增强区间内生成的总图片数量。该数值不固定,请根据原始样本量和训练需求灵活调整。
  • 标注集 (NG):存放该类缺陷样本的图片及其对应的 JSON 标注文件。
  • 负样本 (OK):存放该类正常样本的图片及其对应的 JSON 文件。

注意:文件夹隔离

在进行本地训练打包时,OK 样本与 NG 样本必须存放在不同的文件夹中,严禁混放,以免影响模型对特征的提取。


4. 执行生成

  1. 完成上述镜像、角度及参数配置。
  2. 点击 [生成] 按钮。
  3. 在弹出的对话框中选择用于存放训练集的目标文件夹。
  4. 等待进度条完成,系统将输出最终用于模型训练的打包文件。

算法建议

  • 如果您的检测目标具有旋转对称性(如轴承、密封圈),开启 “旋转一周” 能显著提升模型性能。
  • 对于光环境不稳定的工位,适当增加 “色度/对比度” 的变换幅度,可以有效减少由于光照引起的误检。