在使用PIL讀取圖像又保存,發現有些圖像是上下顛倒的,這可真坑 。
從網上沒有搜到爲什麼,好在熱心的網友告訴我:
pil不讀exif信息,所以反而它讀的是最真實的,cv2會自動讀exif信息做旋轉,這個大坑前段時間害慘我了
原來opencv讀圖像會根據exif信息自動旋轉,用windows下的軟件看圖片時也是如此。
我寫了兩個函數,從numpy.array和PIL.Image之間進行轉換,權當參考:
from PIL import Image
import numpy as np
import cv2
# 從opencv讀進來的圖片轉爲PIL的Image對象
def cv2pil(img):
shape = img.shape
if len(shape) == 3:
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
return Image.fromarray(img)
# 從PIL.Image 轉爲opencv中使用的numpy.array
def pil2cv(img_pil):
img_np = np.array(img_pil)
shape = img_np.shape
if len(shape) == 3:
img_np = cv2.cvtColor(img_np, cv2.COLOR_RGB2BGR)
return img_np