導入

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

可憐白雪曲,未遇知音人。恓惶戎旅下,蹉跎淮海濱。 澗樹含朝雨,山鳥哢餘春。我有一瓢酒

查看原文 >>
相關文章