PIL读取具有旋转属性的图片

最新推荐文章于 2024-02-24 20:54:53 发布

原创最新推荐文章于 2024-02-24 20:54:53 发布 · 282 阅读

7 ·

CC 4.0 BY-SA版权

遥感图像处理与深度学习专栏收录该内容

22 篇文章

订阅专栏

本文介绍如何使用Python解析图片EXIF信息，根据Orientation字段自动调整手机拍摄照片的旋转方向，确保正确显示。

部署运行你感兴趣的模型镜像

手机拍的一般都有旋转属性，直接读不行，换成下面的

import PIL
import PIL.ExifTags
import PIL.Image
import PIL.ImageOps
def apply_exif_orientation(image):
    try:
        exif = image._getexif()
    except AttributeError:
        exif = None

    if exif is None:
        return image

    exif = {
        PIL.ExifTags.TAGS[k]: v
        for k, v in exif.items()
        if k in PIL.ExifTags.TAGS
    }

    orientation = exif.get("Orientation", None)

    if orientation == 1:
        # do nothing
        return image
    elif orientation == 2:
        # left-to-right mirror
        return PIL.ImageOps.mirror(image)
    elif orientation == 3:
        # rotate 180
        return image.transpose(PIL.Image.ROTATE_180)
    elif orientation == 4:
        # top-to-bottom mirror
        return PIL.ImageOps.flip(image)
    elif orientation == 5:
        # top-to-left mirror
        return PIL.ImageOps.mirror(image.transpose(PIL.Image.ROTATE_270))
    elif orientation == 6:
        # rotate 270
        return image.transpose(PIL.Image.ROTATE_270)
    elif orientation == 7:
        # top-to-right mirror
        return PIL.ImageOps.mirror(image.transpose(PIL.Image.ROTATE_90))
    elif orientation == 8:
        # rotate 90
        return image.transpose(PIL.Image.ROTATE_90)
    else:
        return image