Python使用openCV获得图片的h,w,c

这里获得的通道，可以是单通道（灰度图），也可以是3通道彩色图

原来的思路是使用img.shape直接获得，后来测试的时候发现单通到的会报错，get到的数据量不够

就想到try和catch来实现 

def read_img(img_path, c=3):
    img = cv2.imread(img_path)
    if c == 1:
        img = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
    return img


def get_shape_test(img_path):
    img = read_img(img_path, 1)
    # print(img)
    print(cl_img2(img))


def cl_img2(img):
    c = 1
    try:
        h, w, c = img.shape
    except Exception as e:
        print(e)
        h, w = img.shape
    return h, w, c

if __name__ == '__main__':
    get_shape_test('1.jpg')