PIL(pillow)读取图片自动旋转的问题

PIL(pillow)读取图片自动旋转的问题

最近在处理一批手机拍摄的图片,使用pillow的读取的时候发现图像是旋转的,导致和标注框对不上,训练模型直接崩了。

查找资料发现大多数的图像查看软件都是会根据图像的exif信息自动旋转图片的,所以在mac上打开是正的,pillow读取的却是原图像。

方法1:改用opencv读取:

import cv2
from PIL import Image

src = "xxx.jpg"
tgt = "xxx-new.jpg"

a = Image.open(src)
print(a.size)

b = cv2.imread(src)
cv2.imwrite(tgt, b)

a = Image.open(tgt)
print(a.size)

输出:

(3264, 1468)
(1468, 3264)

方法2:用ImageOps.exif_transpose()纠正

import cv2
from PIL import Image, ImageOps
src = "xxx.jpg"
a = Image.open(src)
print(a.size)
a = ImageOps.exif_transpose(a)
print(a.size)

输出:

(3264, 1468)
(1468, 3264)

参考:
https://stackoverflow.com/questions/4228530/pil-thumbnail-is-rotating-my-image

你可能感兴趣的:(踩坑,pillow,opencv,计算机视觉,python)