问题：mmdet.Albu数据增强不生效？

Question

在使用 MMDetection 中的 `Albu` 数据增强模块时，常常会遇到配置正确但数据增强不生效的问题。常见原因之一是未正确安装 `albumentations` 库或版本不兼容，导致配置被静默忽略。此外，在数据流水线配置中，`Albu` 变换应置于 `Normalize` 之前、`DefaultFormatBundle` 之后，否则会影响增强效果。还有可能是未将 `img_fields` 和 `bbox_params` 正确设置，导致图像与标注不同步变换。建议检查配置文件、打印训练流水线确认顺序，并通过可视化中间结果验证增强是否实际执行。

Answer 1

解决 MMDetection 中 Albu 数据增强不生效的问题

1. 环境依赖与安装验证

使用 Albu 模块的前提是正确安装 albumentations 库。如果未安装或版本不兼容，MMDetection 会静默忽略该配置项。

• 检查是否安装：

pip show albumentations

• 推荐安装命令：

pip install albumentations==0.5.2

不同版本的 albumentations 在 API 上可能存在差异，建议根据所使用的 MMDetection 版本选择兼容版本。

2. 配置顺序：流水线中的位置至关重要

在数据增强流水线中，Albu 的位置应满足以下原则：

变换名称	位置要求
Normalize	应在 Albu 之后执行
DefaultFormatBundle	应在 Albu 之前执行

错误示例：

train_pipeline = [
    dict(type='LoadImageFromFile'),
    dict(type='LoadAnnotations', with_bbox=True),
    dict(type='DefaultFormatBundle'),
    dict(type='Albu', transforms=[...]),  # ❌ 错误位置
    dict(type='Normalize', ...),
]

正确顺序示例：

train_pipeline = [
    dict(type='LoadImageFromFile'),
    dict(type='LoadAnnotations', with_bbox=True),
    dict(type='DefaultFormatBundle'),  # ✅ 放在 Albu 前面
    dict(type='Albu', transforms=[...]),  # ✅ 正确位置
    dict(type='Normalize', ...),        # ✅ 放在 Albu 后面
]

3. 图像与标注同步：img_fields 与 bbox_params 设置

为了确保图像和边界框（bbox）同步变换，必须在 Albu 配置中明确指定 img_fields 和 bbox_params。

dict(
    type='Albu',
    transforms=[
        dict(type='ShiftScaleRotate', shift_limit=0.0625, scale_limit=0.1, rotate_limit=45, p=0.5)
    ],
    img_fields=['img'],         # ✅ 必须设置
    bbox_params=dict(            # ✅ 必须设置
        type='BboxParams',
        format='pascal_voc',
        label_fields=['gt_labels']
    )
)

如果不设置这两个参数，可能导致图像被增强而标注框未同步变化，从而影响训练效果。

4. 调试与验证方法

即使配置看起来正确，也应通过以下方式验证数据增强是否真正生效：

1. 打印训练流水线：查看实际加载的 pipeline 是否包含 Albu 模块。
2. 可视化中间输出：在训练前添加调试代码，将增强后的图像和标注框保存下来查看是否正确变换。

from mmdet.datasets.pipelines import Compose

pipeline = Compose(train_pipeline)
data = dataset[0]
augmented = pipeline(data)
import cv2
cv2.imwrite('augmented.jpg', augmented['img'].numpy())

5. 总结性流程图

以下是使用 Albu 数据增强模块时的典型问题排查流程：

graph TD
    A[开始] --> B{albumentations 是否安装?}
    B -- 是 --> C{Albu 位置是否正确?}
    C -- 正确 --> D{img_fields & bbox_params 是否设置?}
    D -- 是 --> E[进行训练]
    D -- 否 --> F[设置必要字段]
    C -- 否 --> G[调整到 DefaultFormatBundle 后、Normalize 前]
    B -- 否 --> H[安装 albumentations 并确认版本]