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