gistfile1.txt

def get_multi_masks(img):
    '''
    Get connected components (multi-masks)
    args: 
        img: np.ndarray
    return:
        np.ndarray with same dimension as the input
    '''

    # Threshold the image to make it become either 0 or 1
    _, thresh = cv2.threshold(img, 10, 255, cv2.THRESH_BINARY)
    # The skimage label function find the connected component. 
    # The pixels of the first component is marked as 1, 
    # the pixels of the second one is marked as 2, and so on and so forth.
    
    return label(thresh)