在做深度學習的時候,我們首先會用到python PIL模塊中的convert函數將原始圖片(例如png)轉化為對應的像素值,再將像素值轉化成tensor之后進行模型的訓練。
1、安裝PIL
1 pip install pillow 2 3 conda install pillow
2、使用方式
1 from PIL import Image 2 3 input_img = Image.open('input.png') 4 5 output_img = input_img.convert('1') 6 7 output_img.save('output.png')
3、參數
convert中可設置轉換模式,介紹比較常用的三種模式:
(1)RGB模式
RGB色彩模式是工業界的一種顏色標准,是通過對紅(R)、綠(G)、藍(B)三個顏色通道的變化以及它們相互之間的疊加來得到各式各樣的顏色的,每個通道的值在0~255之間。
使用示例:
1 output_img = input_img.convert('RGB')
(2)1模式
轉化為為二值圖像,非黑即白,每個像素用8個bit表示,0表示黑,255表示白。
使用示例:
output_img = input_img.convert('1')
(3)L模式
轉化為為灰色圖像,每個像素用8個bit表示,0表示黑,255表示白,0~255代表不同的灰度。需要注意的是,在PIL中,RGB是通過以下公式轉化為L的:
L = R * 299/1000 + G * 587/1000 + B * 114/1000
使用示例:
output_img = input_img.convert('L')