>>> import numpy as np
>>> import albumentations as A
>>> import cv2
>>>
>>> # Prepare primary data
>>> primary_image = np.random.randint(0, 256, (100, 100, 3), dtype=np.uint8)
>>> primary_mask = np.random.randint(0, 2, (100, 100), dtype=np.uint8)
>>> primary_bboxes = np.array([[10, 10, 40, 40], [50, 50, 90, 90]], dtype=np.float32)
>>> primary_bbox_classes = [1, 2]
>>> primary_keypoints = np.array([[25, 25], [75, 75]], dtype=np.float32)
>>> primary_keypoint_classes = ['eye', 'nose']
>>>
>>> # Prepare additional images for mosaic
>>> additional_image1 = np.random.randint(0, 256, (100, 100, 3), dtype=np.uint8)
>>> additional_mask1 = np.random.randint(0, 2, (100, 100), dtype=np.uint8)
>>> additional_bboxes1 = np.array([[20, 20, 60, 60]], dtype=np.float32)
>>> additional_bbox_classes1 = [3]
>>> additional_keypoints1 = np.array([[40, 40]], dtype=np.float32)
>>> additional_keypoint_classes1 = ['mouth']
>>>
>>> additional_image2 = np.random.randint(0, 256, (100, 100, 3), dtype=np.uint8)
>>> additional_mask2 = np.random.randint(0, 2, (100, 100), dtype=np.uint8)
>>> additional_bboxes2 = np.array([[30, 30, 70, 70]], dtype=np.float32)
>>> additional_bbox_classes2 = [4]
>>> additional_keypoints2 = np.array([[50, 50], [65, 65]], dtype=np.float32)
>>> additional_keypoint_classes2 = ['eye', 'eye']
>>>
>>> additional_image3 = np.random.randint(0, 256, (100, 100, 3), dtype=np.uint8)
>>> additional_mask3 = np.random.randint(0, 2, (100, 100), dtype=np.uint8)
>>> additional_bboxes3 = np.array([[5, 5, 45, 45]], dtype=np.float32)
>>> additional_bbox_classes3 = [5]
>>> additional_keypoints3 = np.array([[25, 25]], dtype=np.float32)
>>> additional_keypoint_classes3 = ['nose']
>>>
>>> # Create metadata for additional images - structured as a list of dicts
>>> # Note: label field names must match those in bbox_params/keypoint_params
>>> mosaic_metadata = [
...     {
...         'image': additional_image1,
...         'mask': additional_mask1,
...         'bboxes': additional_bboxes1,
...         'bbox_classes': additional_bbox_classes1,
...         'keypoints': additional_keypoints1,
...         'keypoint_classes': additional_keypoint_classes1
...     },
...     {
...         'image': additional_image2,
...         'mask': additional_mask2,
...         'bboxes': additional_bboxes2,
...         'bbox_classes': additional_bbox_classes2,
...         'keypoints': additional_keypoints2,
...         'keypoint_classes': additional_keypoint_classes2
...     },
...     {
...         'image': additional_image3,
...         'mask': additional_mask3,
...         'bboxes': additional_bboxes3,
...         'bbox_classes': additional_bbox_classes3,
...         'keypoints': additional_keypoints3,
...         'keypoint_classes': additional_keypoint_classes3
...     }
... ]
>>>
>>> # Create the transform with Mosaic
>>> transform = A.Compose([
...     A.Mosaic(
...         grid_yx=(2, 2),
...         target_size=(200, 200),
...         cell_shape=(120, 120),
...         center_range=(0.4, 0.6),
...         fit_mode="cover",
...         p=1.0
...     ),
... ], bbox_params=A.BboxParams(format='pascal_voc', label_fields=['bbox_classes']),
...    keypoint_params=A.KeypointParams(format='xy', label_fields=['keypoint_classes']))
>>>
>>> # Apply the transform
>>> transformed = transform(
...     image=primary_image,
...     mask=primary_mask,
...     bboxes=primary_bboxes,
...     bbox_classes=primary_bbox_classes,
...     keypoints=primary_keypoints,
...     keypoint_classes=primary_keypoint_classes,
...     mosaic_metadata=mosaic_metadata  # Pass the metadata using the default key
... )
>>>
>>> # Access the transformed data
>>> mosaic_image = transformed['image']                    # Combined mosaic image
>>> mosaic_mask = transformed['mask']                      # Combined mosaic mask
>>> mosaic_bboxes = transformed['bboxes']                  # Combined and repositioned bboxes
>>> mosaic_bbox_classes = transformed['bbox_classes']      # Combined bbox labels from all images
>>> mosaic_keypoints = transformed['keypoints']            # Combined and repositioned keypoints
>>> mosaic_keypoint_classes = transformed['keypoint_classes']  # Combined keypoint labels
Parameters

Bounding Box Parameters

Keypoint Parameters
