python中cv2模塊imread函數

導入

>> import cv2
>> import numpy as np

讀圖片

>> image_arr = cv2.imread('file_path')

灰度圖擴展成彩色圖

能夠經過圖片的channel判斷是不是灰度圖。若是須要能夠將灰度圖擴展到RGB的彩色圖（複製灰度圖的數據到各通道）

>> if image_arr.shape[2] == 1:
      image_arr_rgb = cv2.cvtColor(image_arr, cv2.COLOR_GRAY2RGB)

彩色圖像素存儲格式

imread 讀的彩色圖按照BGR像素存儲，若是轉換成RGB則須要用cvtColor函數進行轉換

>> image_arr_rgb = cv2.cvtColor(image_arr, cv2.COLOR_BGR2RGB)

圖片size存儲格式

imread 讀的圖片按照 H,W,C 格式存儲

>> image_arr_rgb.shape
(H, W, C)

H,W,C格式轉換到C,H,W格式

>> image_arr_rgb_chw = np.transpose(image_arr_rgb, (2,0,1))