五、 额外部分:数据增强
功能说明
- 目前该功能并未集成在 「ATSVisionOps」 当中,属于另外单独的程序功能,但不排除后期我会将其功能集成在 「ATSVisionOps」 当中。
通过数据增强功能,您可以基于现有样本生成更多样化的训练数据,从而提升模型在工业现场的抗干扰能力和检测精度。
1. 开启功能¶
打开 训练数据打包工具 。

在主界面双击进入 「目标检测数据集」 应用程序。
2. 增强算子说明¶

请根据您的产品特征(如圆形、方形、表面纹理等)选择合适的镜像或变换方式:
| 增强模式 | 核心逻辑 | 适用场景 |
|---|---|---|
| 横向镜像 | 以 Y 轴为中轴线进行左右翻转。 | 适用于框标图,如区分产品的内外侧。 |
| 纵向镜像 | 以 X 轴为中轴线进行上下翻转。 | 适用于框标图,增加位置鲁棒性。 |
| 旋转一周 | 以图像圆心为原点进行旋转,角度可自由调整。 | 推荐用于点标图,常用于圆形表面缺陷检测。 |
| 色度/对比度 | 在限定区间内随机调整图像色彩与明暗。 | 默认设置即可,用于模拟产线光照变化。 |
3. 参数配置¶
在界面右侧或下方配置以下关键参数:
- 图数:指在设定的增强区间内生成的总图片数量。该数值不固定,请根据原始样本量和训练需求灵活调整。
- 标注集 (NG):存放该类缺陷样本的图片及其对应的 JSON 标注文件。
- 负样本 (OK):存放该类正常样本的图片及其对应的 JSON 文件。
注意:文件夹隔离
在进行本地训练打包时,OK 样本与 NG 样本必须存放在不同的文件夹中,严禁混放,以免影响模型对特征的提取。
4. 执行生成¶
- 完成上述镜像、角度及参数配置。
- 点击 [生成] 按钮。
- 在弹出的对话框中选择用于存放训练集的目标文件夹。
- 等待进度条完成,系统将输出最终用于模型训练的打包文件。
算法建议
- 如果您的检测目标具有旋转对称性(如轴承、密封圈),开启 “旋转一周” 能显著提升模型性能。
- 对于光环境不稳定的工位,适当增加 “色度/对比度” 的变换幅度,可以有效减少由于光照引起的误检。