Transform image with bounding box albumentations

bounding_box_albumentations.py

# Create bounding boxes from mask with cv2
import cv2
mask = cv2.cvtColor(mask, cv2.COLOR_BGR2GRAY)
bboxes = cv2.boundingRect(cv2.findNonZero(mask))

# Augment image and bounding box
augmented_boxes = aug_pipeline(image = image, bboxes = [bboxes])

# Access augmented image and bounding box
image_aug = augmented_boxes['image'][0]
box_aug = augmented_boxes['bboxes'][0]