utils/segmentation.py

import cv2


def get_connected_regions(binary):

    num_labels, labels = cv2.connectedComponents(
        binary
    )

    return num_labels, labels