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))
可憐白雪曲,未遇知音人。恓惶戎旅下,蹉跎淮海濱。 澗樹含朝雨,山鳥哢餘春。我有一瓢酒
查看原文 >>