image helper

img_util.py

def color_kinds(img):
    color_set = set()
    if np.ndim(img) == 2:
        channel_num = 1
    elif np.ndim(img) >= 3:
        channel_num = np.size(img, np.ndim(img) - 1)
    else:
        print('unknow img format')
        return color_set

    print('img real channel:{}'.format(channel_num))
    for i in range(np.shape(img)[0]):
        for j in range(np.shape(img)[1]):
            color_str = ''
            if channel_num == 1:
                color_str = str(img[i][j])
            else:
                for c in range(channel_num):
                    color_str = color_str + ' ' + str(img[i][j][c])
            color_set.add('({})'.format(color_str))
    return color_set