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 次增强组成。

注意：该层可以在 tf.data 或 grain 管道中使用（无论您使用的是哪个后端），且是安全的。

参考文献

• AugMix 论文 - 官方代码

参数

• 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：整数。用于创建随机种子。