python OpenCV 图像通道数判断

import cv2

# 参数1：文件路径
# 参数2：读取方式
# 常用几种模式：
# 	灰度图：cv2.IMREAD_GRAYSCALE
#	BGR三通道彩色图：cv2.IMREAD_COLOR
#	原始通道（自动判断：单通道/三通道/四通道等）：cv2.IMREAD_UNCHANGED
img = cv2.imread('图像文件路径（全英文）', cv2.IMREAD_COLOR)

# 参数1：文件路径
# 参数2：待保存的图像（BGR格式的uint8 ndarray类型）
# 参数3【可选】：压缩模式以及质量等
# 常用几种模式：
# 	灰度图：cv2.IMREAD_GRAYSCALE
#	BGR三通道彩色图：cv2.IMREAD_COLOR
#	原始通道（自动判断：单通道/三通道/四通道等）：cv2.IMREAD_UNCHANGED
img_saved = cv2.imwrite('图像输出路径（含图像名）', img)

import cv2
import numpy as np

# cv2.imdecode()：从指定的内存缓存中读取数据,并把数据转换成图像格式
src_img_path = './image.png'
img=cv2.imdecode(np.fromfile(src_img_path, dtype=np.uint8),cv2.IMREAD_UNCHANGED)

# cv2.imencode()：将图片格式转换成流数据，写入内存缓存中，可以再调用.tofile()写入到磁盘中
out_img_path = './image.jpg'
extension = '.jpg'
cv2.imencode(extension,img)[1].tofile(out_img_path)

img_rgb = cv2.cvtColor(img,cv2.COLOR_BGR2RGB)

img_gray=cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)

if img.ndim == 2:
	print('图像是单通道灰度图')
else:
	print('图像包含多个通道')

def get_n_channel(img):
	if img.ndim == 2:
		print('通道数：1')
		return 1
	else:
		print('图像包含多个通道')
		return img.shape[2]