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
'''