sep_labels.py 文件源码

def get_separate_labels(label_img):
    # 64-bit encoding
    dtype = label_img.dtype
    if dtype == np.uint8:
        w = 8
    elif dtype == np.uint16:
        w = 16
    else:
        raise Exception('Unknown dtype: "{}"'.format(dtype))
    l64 = label_img.astype('uint64')
    # Single channel mapping
    if len(l64.shape) == 3:
        l64i = ((l64[:, :, 0] << 2 * w) + (l64[:, :, 1] << w) + l64[:, :, 2])
    else:
        l64i = l64
    colors = np.unique(l64i)
    segmentations = []
    colors_all = []
    for c in colors:
        if c != 0:
            segmentation = (l64i == c).astype('uint8')
            segmentations.append(segmentation)
            colors_all.append(c)
    return segmentations, colors_all