AugMix 层

[源代码]

AugMix 类

keras.layers.AugMix(
    value_range=(0, 255),
    num_chains=3,
    chain_depth=3,
    factor=0.3,
    alpha=1.0,
    all_ops=True,
    interpolation="bilinear",
    seed=None,
    data_format=None,
    **kwargs
)

执行 AugMix 数据增强技术。

AugMix 旨在生成具有多样性且同时保留图像语义和局部统计信息的图像。在增强过程中，对批次中的所有图像以 num_chains 种不同的方式应用相同的增强，每个链包含 chain_depth 个增强。

参数

• value_range：输入图像的值范围。表示为一个两元素的元组 `(low, high)`。这通常是 `(0, 1)` 或 `(0, 255)`，具体取决于您的预处理管道设置。
• num_chains：一个整数，表示要混合的不同链的数量，默认为 3。
• chain_depth：一个整数，表示在每个链中应用的最大变换次数。每个链中的实际变换次数将从范围 `[0, `chain_depth`]` 中随机采样。默认为 3。
• factor：增强的强度，作为 0 到 1 之间的归一化值。默认为 0.3。
• alpha：一个浮点值，用作 Beta 和 Dirichlet 分布的概率系数，默认为 1.0。
• all_ops：使用所有操作（包括 random_brightness、random_color_degeneration、random_contrast 和 random_sharpness）。默认为 True。
• interpolation：用于调整大小操作的插值方法。选项包括 "nearest"、"bilinear"。默认为 "bilinear"。
• seed：整数。用于创建随机种子。

参考

• AugMix 论文 - 官方代码