RandAugmentΒΆ

RandAugment performs the Rand Augment operation on input images.

Abstract Signature:

RandAugment(num_ops: int = 2, magnitude: int = 9)

PyTorch

API: torchvision.transforms.RandAugment
Strategy: Direct Mapping

Keras

API: keras.layers.RandAugment
Strategy: Direct Mapping

TensorFlow

API: tf.keras.layers.RandAugment
Strategy: Direct Mapping

Apple MLX

API: β€”
Strategy: Custom / Partial

Flax NNX

API: β€”
Strategy: Custom / Partial

PaxML / Praxis

API: β€”
Strategy: Custom / Partial