Image Augmentations

The augmentation operators, grouped by what they change. All of them accept (B, C, H, W) float tensors in [0, 1] ((B, C, D, H, W) for the 3D transforms), sample independent parameters per batch element, and can be combined with AugmentationSequential.

2D intensity transforms	2D transforms that change pixel values but not their positions: color jitter, blur, noise, illumination, JPEG, weather effects.
2D geometric transforms	2D transforms that move pixels: crops, flips, affine, perspective, elastic, thin-plate-spline, erasing.
2D mix transforms	2D transforms that combine several images: CutMix, MixUp, Mosaic, Jigsaw, Transplantation.
3D transforms	Geometric, intensity and mix transforms for volumetric (B, C, D, H, W) data.