Search papers, labs, and topics across Lattice.
AlbumentationsX introduces a unified augmentation pipeline that ensures consistent transformations across images and their annotations, addressing the common issue of misalignment caused by separate random changes. By consolidating the transform list, probabilities, and random seed into a single Compose object, it guarantees that all components of a training example are modified in sync. The library's design allows for easy integration of custom transforms and facilitates reproducibility by enabling users to save and revisit specific augmentation configurations.
Misalignment in data augmentation can lead to significant training errors, but AlbumentationsX ensures that images and their annotations are always transformed together, preserving label integrity.
Augmentation can corrupt a training example when an image and its annotations receive different random changes. A crop must use the same coordinates for the image, mask, boxes, keypoints, stereo views, video frames, or volume. Code paths that choose these values separately can silently misalign the data. AlbumentationsX keeps the transform list, probabilities, annotation settings, and random seed in one Compose object. Each call chooses random values once and applies them to every supported part of the training example. The library keeps each object's mask, box, and label together and lets projects add their own transforms. It can also save the pipeline definition, show what happened in one call, and run that call again. The examples place Compose after files have been decoded into arrays and before PyTorch groups examples into a batch. AlbumentationsX executes the declared transforms. Practitioners still decide whether a flip, crop, color change, or other operation preserves the correct label for their task.
