mirror of https://github.com/alibaba/EasyCV.git
21 lines
521 B
Python
21 lines
521 B
Python
# Copyright (c) Alibaba, Inc. and its affiliates.
|
|
from PIL import ExifTags
|
|
|
|
|
|
def exif_size(img):
|
|
# Get orientation exif tag
|
|
for orientation in ExifTags.TAGS.keys():
|
|
if ExifTags.TAGS[orientation] == 'Orientation':
|
|
break
|
|
|
|
# Returns exif-corrected PIL size
|
|
s = img.size # (width, height)
|
|
|
|
rotation = dict(img._getexif().items())[orientation]
|
|
if rotation == 6: # rotation 270
|
|
s = (s[1], s[0])
|
|
elif rotation == 8: # rotation 90
|
|
s = (s[1], s[0])
|
|
|
|
return s
|