python-改变图片通道排列

'''  PIL 中的convert() 也有着改变通道排列的效果，期待之后补充'''

import cv2
from PIL import Image
import numpy as np
import matplotlib.pyplot as plt



img_path = '../data/a.jpg'
print("="*15,"Image.open() 读取图像","="*15)
img = Image.open(img_path)
print("Image.open() 读取图片，type(img) :",type(img))    # <class 'PIL.JpegImagePlugin.JpegImageFile'>
print("Image.open() 读取图片，img.size (没有shape属性) :",img.size)   # (328, 260)
print("Image.open() 读取图片，img.mode :",img.mode)      # RGB
# Image.open() 读取图片，没有shape属性，只有size（高，宽）
plt.subplot(221)
plt.imshow(img)

print("="*15,"cv2.imread() 读取图像","="*15)
img_cv2 = cv2.imread(img_path)
print("cv2.imread() 读取图片, type(img_cv2)：", type(img_cv2))       # <class 'numpy.ndarray'>
print("cv2.imread() 读取图片(高，宽，通道(B,G,R))：",img_cv2.shape)   # (260, 328, 3)
# 将通道的排列放在宽高之前
img = np.transpose(img_cv2, (2,0,1))
print("通过np.transpose 函数将通道放在宽高之前: ",img.shape)           # (3, 260, 328)

# 显示原图（BGR排序），OpenCV 默认读取图片的数据为 (高，宽，通道(B，G，R))
plt.subplot(223)
plt.imshow(img_cv2)
# 尝试转换通道排列顺序，并显示转换后图像（RGB）
img_RGB = img_cv2[:,:,(2,1,0)]  # 改变通道的排列顺序
plt.subplot(224)
plt.imshow(img_RGB)

img_RGB2 = img_cv2[:,:,::-1]    # 可以实现与上一行代码相同的效果
plt.subplot(222)
plt.imshow(img_RGB2)
print("OpenCV 默认读入图片颜色通道排列为 BGR,可以通过 img_RGB = img_cv2[:,:,(2,1,0)] 或 img_RGB2 = img_cv2[:,:,::-1]  将颜色通道排列改为 RGB")

plt.show()

''' 运行结果：
=============== Image.open() 读取图像 ===============
Image.open() 读取图片，type(img) : <class 'PIL.JpegImagePlugin.JpegImageFile'>
Image.open() 读取图片，img.size (没有shape属性) : (328, 260)
Image.open() 读取图片，img.mode : RGB
=============== cv2.imread() 读取图像 ===============
cv2.imread() 读取图片, type(img_cv2)： <class 'numpy.ndarray'>
cv2.imread() 读取图片(高，宽，通道(B,G,R))： (260, 328, 3)
通过np.transpose 函数将通道放在宽高之前:  (3, 260, 328)
OpenCV 默认读入图片颜色通道排列为 BGR,可以通过 img_RGB = img_cv2[:,:,(2,1,0)] 或 img_RGB2 = img_cv2[:,:,::-1]  将颜色通道排列改为 RGB
'''