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 论文 - 官方代码