luxonis_ml.data.augmentations.custom

package documentation

Built-in custom Albumentations transforms.

The TRANSFORMATIONS registry contains Luxonis transforms that can be used by AlbumentationsEngine through augmentation configuration records. Registered transforms are referenced by class name in the name field, the same way as standard Albumentations transforms.

Built-in custom transforms include:

  • LetterboxResize for aspect-ratio preserving resize and padding.
  • MixUp for blending pairs of images and compatible labels.
  • Mosaic4 for composing a 2×2 image mosaic.
  • Symmetric keypoint flips and transposition helpers for keypoint tasks.

User-defined transforms can be registered with TRANSFORMATIONS.register and then used by name in loader augmentation configuration.

TRANSFORMATIONS.register(module=CustomTransform)

augmentation_config = [
    {
        "name": "CustomTransform",
        "params": {"p": 1.0},
    },
]
Module letterbox_resize Undocumented
Module mixup Undocumented
Module mosaic Undocumented
Module symmetric_keypoints_flip Undocumented

From __init__.py:

Class HorizontalSymmetricKeypointsFlip Flip images and symmetric keypoints horizontally.
Class LetterboxResize Augmentation that resizes an image with padding to maintain the aspect ratio.
Class MixUp Batch-based augmentation that blends two images together.
Class Mosaic4 Batch-based augmentation that creates a mosaic of four images.
Class TransposeSymmetricKeypoints Transpose images and symmetric keypoints.
Class VerticalSymmetricKeypointsFlip Flip images and symmetric keypoints vertically.
Constant TRANSFORMATIONS Undocumented
TRANSFORMATIONS: Registry[type[A.BasicTransform]] =

Undocumented

Value
Registry('albumentations_transformations')