yolov11使用强数据增强 - 文章 - 开发者社区

picture.image

向AI转型的程序员都关注公众号机器学习AI算法工程

在深度学习的训练中，强数据增强（strong data augmentation）通过对训练数据进行更大幅度的随机变换，增强模型的泛化能力，减少过拟合风险。强数据增强可以包括各种随机的图像变换操作，使得模型能够更好地适应不同的场景和变化。以下是一些常见的强数据增强操作，以及如何在 YOLOv8 训练中使用它们：

一、常见的强数据增强操作

1.1 翻转和旋转：

随机水平翻转、垂直翻转。

随机旋转一定角度（比如±90度），使模型适应不同角度的对象。

1.2 缩放和裁剪：

随机缩放图像的不同部分，让模型适应不同大小的目标。

随机裁剪图像的一部分，或者裁剪成不同的长宽比。

1.3 颜色抖动：

改变亮度、对比度、饱和度和色调。通过这些变换，模型可以适应不同的光照条件。

1.4 模糊和噪声：

添加随机噪声或使用高斯模糊来模拟图像中的噪声，提高模型对低质量图像的鲁棒性。

1.5 仿射变换：

通过平移、缩放、旋转等几何变换来改变图像的形状，使模型适应更多的场景变化。

1.6 随机擦除：

随机地在图像中插入一些黑色或白色的方块，模拟图像中的遮挡物，增强模型的鲁棒性。

1.7 CutMix 或 Mosaic：

这是YOLO系列中特别常用的增强方法：

CutMix：将两张图片切分并合成一张图片。

Mosaic：将四张图片拼接成一张，这样每张图中的对象数量和位置会有变化。

二、在 YOLOv8 中使用强数据增强

YOLOv8 的训练配置文件中已经内置了丰富的数据增强选项。可以通过修改配置文件或者直接在训练命令中调整数据增强策略。以下是一些常见的增强选项及如何配置：

2.1 配置文件修改：

YOLOv8 的 data.yaml 文件中可以直接设置数据增强。你可以根据需求启用或调整增强操作的强度。例如：


          
augment:
          
  flipud: 0.5      # 50% 概率进行垂直翻转
          
  fliplr: 0.5      # 50% 概率进行水平翻转
          
  mosaic: 1.0      # 启用 Mosaic 数据增强
          
  mixup: 0.5       # 启用 Mixup 数据增强
          
  hsv_h: 0.015     # 色调增强，范围为 ±0.015
          
  hsv_s: 0.7       # 饱和度增强，范围为 ±0.7
          
  hsv_v: 0.4       # 亮度增强，范围为 ±0.4
          
  scale: 0.5       # 随机缩放，范围为 ±50%
          
  shear: 0.0       # 随机剪切，设置为 0 禁用
          
  perspective: 0.0 # 随机透视变换，设置为 0 禁用

训练命令中直接设置：可以通过 YOLOv8 训练命令直接启用一些增强选项。例如：

yolo train model=yolov8m-pose.pt data=data.yaml epochs=100 augment=True

这里的 augment=True 会启用默认的数据增强策略。如果你想要更强的增强效果，可以在配置文件中添加 Mosaic、Mixup 等额外增强策略。

2.2 使用自定义的数据增强策略

如果默认的数据增强不够，你也可以自定义更复杂的增强方式，比如：

使用 Albumentations 库：YOLOv8 支持集成 Albumentations，这个库提供了丰富的数据增强功能，可以自定义强数据增强策略。


          
from ultralytics import YOLO
          
from albumentations import Compose, RandomBrightnessContrast, HueSaturationValue, ShiftScaleRotate
          
# 定义强数据增强
          
def custom_augment():
          
    return Compose([
          
        RandomBrightnessContrast(p=0.5),
          
        HueSaturationValue(hue_shift_limit=20, sat_shift_limit=30, val_shift_limit=20, p=0.5),
          
        ShiftScaleRotate(shift_limit=0.1, scale_limit=0.1, rotate_limit=15, p=0.5)
          
    ])
          

          
# 加载模型
          
model = YOLO('yolov8m-pose.pt')
          

          
# 启用自定义数据增强
          
model.train(data='data.yaml', epochs=100, augment=True, custom_augment=custom_augmen

三、总结

强数据增强可以通过组合多种图像变换（翻转、旋转、裁剪、颜色抖动等）实现。在 YOLOv8 中，你可以通过调整 data.yaml 文件中的参数来控制增强的强度，或者使用自定义的增强库（如 Albumentations）来实现更复杂的增强方案。这些操作可以显著提高模型的泛化能力，使其更好地适应复杂的真实场景。

机器学习算法AI大数据技术

搜索公众号添加： datanlp

picture.image