覆盖主要内容
版本: 6.1.0

颜色抖动

组件介绍

“颜色抖动”(Color Jitter) 控件通过随机调整图像的亮度、对比度、饱和度和色相,模拟不同光照条件和成像环境,从而增强模型对颜色变化的鲁棒性与泛化能力。

目前该组件仅支持 ImageNet 格式数据集(即包含 images/ 子目录及 *_list.txt 文件的结构),并会保留原始目录结构。

该控件主要执行以下步骤:

  1. 检查输入是否为 ImageNet 格式;
  2. images/ 目录下每张图像应用颜色抖动,生成带 _jitter 后缀的新图像;
  3. 自动更新所有 *_list.txt 文件(如 train_list.txtval_list.txt 等),将增强图像及其标签(如有)追加写入;
  4. 输出增强后的完整数据集对象。

  • 输入:
    • img:特定格式的图像数据集(含 images/*_list.txt
  • 输出:
    • img:增强后的特定格式数据集(包含原始图像 + _jitter 增强图像)

页面介绍

点击 “颜色抖动”(Color Jitter) 控件可进入参数配置页面,如下图所示:

颜色抖动 参数配置

参数选项

选项说明取值范围样例值
亮度调整亮度扰动强度。实际亮度因子 = 1 ± brightness,在 [1−b, 1+b] 范围内均匀采样。0.0 – 1.00.2
对比度调整对比度扰动强度。实际对比度因子 = 1 ± contrast,在 [1−c, 1+c] 范围内均匀采样。0.0 – 1.00.2
饱和度调整饱和度扰动强度。实际饱和度因子 = 1 ± saturation,在 [1−s, 1+s] 范围内均匀采样。0.0 – 1.00.2
色相调整色相扰动强度。实际色相偏移角度 = ± hue × 180°,在 [−h×180°, +h×180°] 范围内均匀采样(HSV 色彩空间循环偏移)。0.0 – 0.50.1

使用案例

在下图所示的工作流中,使用 “图像 控件加载图像数据,通过 “颜色抖动” 对图像进行颜色增强,最后使用 “保存图像” 保存处理结果。

工作流示意图

案例中加载了自定义水果数据集“颜色抖动”(Color Jitter) 控件参数设置如下:

参数配置示例

分别查看原始图像与增强后图像效果对比:

原始图像

增强后图像