ExcelHome技术论坛

 找回密码
 免费注册

QQ登录

只需一步,快速开始

快捷登录

搜索
EH技术汇-专业的职场技能充电站 妙哉!函数段子手趣味讲函数 Excel服务器-会Excel,做管理系统 效率神器,一键搞定繁琐工作
HR薪酬管理数字化实战 Excel 2021函数公式学习大典 Excel数据透视表实战秘技 打造核心竞争力的职场宝典
让更多数据处理,一键完成 数据工作者的案头书 免费直播课集锦 ExcelHome出品 - VBA代码宝免费下载
用ChatGPT与VBA一键搞定Excel WPS表格从入门到精通 Excel VBA经典代码实践指南
查看: 232|回复: 0

[分享] 将D盘根目录图片识别倒顺方向,旋转成正向然后从中间切开,保存到新目录

[复制链接]

TA的精华主题

TA的得分主题

发表于 2024-6-3 23:36 | 显示全部楼层 |阅读模式
import os
from PIL import Image, ExifTags
import pytesseract


# 设置Tesseract的路径,如果已添加到系统路径中则可以省略
# pytesseract.pytesseract.tesseract_cmd = r'C:\Program Files\Tesseract-OCR\tesseract.exe'

def create_directory(directory_path):
    if not os.path.exists(directory_path):
        os.makedirs(directory_path)


def correct_image_orientation(image):
    try:
        for orientation in ExifTags.TAGS.keys():
            if ExifTags.TAGS[orientation] == 'Orientation':
                break
        exif = image._getexif()
        if exif is not None:
            orientation = exif.get(orientation, 1)
            if orientation == 3:
                image = image.rotate(180, expand=True)
            elif orientation == 6:
                image = image.rotate(270, expand=True)
            elif orientation == 8:
                image = image.rotate(90, expand=True)
    except (AttributeError, KeyError, IndexError):
        pass
    return image


def split_image(image, save_path_left, save_path_right):
    width, height = image.size
    left_image = image.crop((0, 0, width // 2, height))
    right_image = image.crop((width // 2, 0, width, height))
    left_image.save(save_path_left)
    right_image.save(save_path_right)


def process_images(source_directory, target_directory):
    create_directory(target_directory)
    for filename in os.listdir(source_directory):
        if filename.lower().endswith('.jpg'):
            file_path = os.path.join(source_directory, filename)
            image = Image.open(file_path)
            image = correct_image_orientation(image)

            # 使用Tesseract识别图像中的文本方向
            ocr_result = pytesseract.image_to_osd(image)
            if 'Rotate: 90' in ocr_result:
                image = image.rotate(-90, expand=True)
            elif 'Rotate: 180' in ocr_result:
                image = image.rotate(-180, expand=True)
            elif 'Rotate: 270' in ocr_result:
                image = image.rotate(-270, expand=True)

            base_name = os.path.splitext(filename)[0]
            save_path_left = os.path.join(target_directory, f"{base_name}_LEFT.jpg")
            save_path_right = os.path.join(target_directory, f"{base_name}_RIGHT.jpg")

            split_image(image, save_path_left, save_path_right)


source_directory = "D:\\"
target_directory = "D:\\SPLITTED PHOTO"
process_images(source_directory, target_directory)


您需要登录后才可以回帖 登录 | 免费注册

本版积分规则

手机版|关于我们|联系我们|ExcelHome

GMT+8, 2024-11-21 18:18 , Processed in 0.034254 second(s), 9 queries , Gzip On, MemCache On.

Powered by Discuz! X3.4

© 1999-2023 Wooffice Inc.

沪公网安备 31011702000001号 沪ICP备11019229号-2

本论坛言论纯属发表者个人意见,任何违反国家相关法律的言论,本站将协助国家相关部门追究发言者责任!     本站特聘法律顾问:李志群律师

快速回复 返回顶部 返回列表