跳到主要内容

数据增强

数据增强通过对已有图片做几何和色彩变换,凭空「变」出更多训练样本。样本越多样,模型见过的场景就越多,泛化到陌生环境的能力也就越强。这一页讲怎么在 VortLab 里给检测数据集跑增强。

适用范围

仅适用于检测项目
  • 数据增强当前只支持目标检测(detect)类型的数据集。
  • 不支持实例分割(segment)、旋转框(obb)、姿态关键点(pose)等其他任务类型。
  • 关键点数据集请交给 Ultralytics 训练内部的增强去处理(通过 flip_idx),平台 OpenCV 增强器尚未实现关键点感知变换。
  • 对不支持的数据集执行增强,界面会直接禁用按钮,后端也会返回错误,避免标签数据丢失或损坏。

入口

数据增强入口在项目工作区的数据集详情页,点击「数据增强」标签页进入。全局「数据集」列表是只读资产总览,不提供增强操作。

操作步骤

  1. 进入项目工作区的数据集详情页

    打开对应项目,从左侧导航进入「数据集」,打开这个项目绑定的数据集详情页。

  2. 点击「数据增强」标签页

    在详情页顶部的标签栏选择「数据增强」,进入增强配置界面。左侧是参数配置卡片,右侧是结果展示区。

  3. 设置扩增倍数

    在「扩增倍数」里填一个数字,范围 1 到 10。比如填 2,表示把原始数据集扩充为原来的两倍。

  4. 选择增强变换

    勾选想启用的变换类型,至少选一项。每项的默认值和参数见下方字段表。

  5. 点击「开始增强」

    确认后点底部「开始增强」。任务在后台异步执行,右侧会显示任务 ID、输出目录和进度,状态在「等待中」「处理中」「已完成」之间切换。

  6. 查看增强结果

    任务完成后,右侧展示三张统计卡片:原始图像数、新生成的增强图像数、合并后的总图像数。下方还会列出每种变换各生成了多少张。

数据增强标签页
图1:数据增强标签页(左侧配置,右侧结果)

配置字段

「扩增倍数」控制总量,下面五项开关控制具体做哪些变换。开关关掉的那项就不参与本次增强。

字段类型必填默认值说明
扩增倍数数字必填2原始数据集扩充为原来的几倍,范围 1 到 10。倍数越大生成样本越多,耗时也越长。
翻转开关可选开启水平翻转和垂直翻转。默认开启,几乎适用于所有检测场景。
旋转开关可选开启按 90°、180°、270° 三个角度旋转。默认开启,边界框会同步变换。
添加噪声开关可选关闭添加高斯噪声,模拟摄像头或低光环境产生的噪点。默认关闭。
亮度和对比度变化开关可选开启亮度在 -20 到 20 之间、对比度在 0.8 到 1.2 之间随机调整,模拟不同光照条件。默认开启。
透视变换开关可选关闭对图像做透视形变,模拟不同拍摄角度。强度固定为 0.2,默认关闭。

支持的变换类型

支持的变换类型
  • 翻转:水平翻转、垂直翻转,最常用的扩充手段。
  • 旋转:90° / 180° / 270° 旋转,边界框坐标同步更新。
  • 噪声添加:高斯噪声,提升模型对低画质输入的容忍度。
  • 亮度 / 对比度变化:随机调整明暗和对比,覆盖不同光照场景。
  • 透视变换:改变观察视角,增加角度多样性。
增强结果统计
图2:增强完成后的统计卡片与变换分布

为什么要做增强

增强的作用
  • 增加训练数据的多样性,让模型见过更多变体,降低过拟合风险。
  • 提升模型对不同光照、角度、尺度场景的鲁棒性。
  • 对数据量较少的数据集尤其有效,少量原图也能撑起一轮训练。
  • 对检测任务,增强后的边界框会跟随图像同步变换,标签不会错位。

与数据集版本的关系

增强后需重新发布版本
  • 增强产物会作为补充样本进入数据准备流程。
  • 训练任务只读取已发布的数据集版本,不会直接读取可变工作区。
  • 如果想让训练用上增强后的数据,需要重新发布一个新版本。
  • 已发布的旧版本内容固定、不受影响,可以放心对比新旧两版的训练效果。

接下来

  • 想了解版本发布、归档、恢复的完整流程,看「数据集与版本」分类下的版本说明。
  • 增强样本准备就绪后,回到项目训练菜单创建训练任务即可。