百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 编程字典 > 正文

「原创文章」python实现PDF转换TXT格式pytesseract详解

toyiye 2024-09-07 20:31 3 浏览 0 评论

pytesseract介绍:

pytesseract是基于Python的OCR工具, 底层使用的是Google的Tesseract-OCR 引擎,支持识别图片中的文字,支持jpeg, png, gif, bmp, tiff等图片格式。本文介绍如何使用pytesseract 实现图片文字识别。

什么是OCR?

OCR(Optical character recognition,光学字符识别)是一种将图像中的手写字或者印刷文本转换为机器编码文本的技术。通过数字方式存储文本数据更容易保存和编辑,可以存储大量数据,比如1G的硬盘可以存储数百万本书。

OCR可以做什么?

OCR技术可以将图片,纸质文档中的文本转换为数字形式的文本。OCR过程一般包括以下步骤:

图像预处理/文本定位/字符分割/字符识别/后处理,最初由惠普开发,后来Google赞助的开源OCR引擎 tesseract 提供了比较精确的文字识别API,支持100多种语言,本文将要介绍的Python库Pytesseract就是基于Tesseract-OCR 引擎。

环境要求:(win环境和liunx均可)

  • Python 3.6+

  • PIL库

  • 安装Google Tesseract OCR

  • 系统:windows/mac/linux,我的系统是Windows10

软件安装:(win环境支持)

Tesseract OCR github地址:https://github.com/tesseract-ocr/tesseract

Windows Tesseract下载地址:https://digi.bib.uni-mannheim.de/tesseract/

Mac和Linux安装方法参考:https://tesseract-ocr.github.io/tessdoc/Installation.html

安装时可以选择需要的语言包:

安装完成后,添加到环境变量PATH中,我的安装路径是:C:\Program Files\Tesseract-OCR

命令行窗口输入:tesseract ,查看是否安装成功。

温馨提示:最好不要安装到C盘因为有些电脑需要管理员权限

安装pytesseract(python环境依赖)

pip安装pytesseract,Python tesseract:https://github.com/madmaze/pytesseract

pip install pytesseract

程序案例:


import fitzfrom PIL import Imageimport pytesseract   def extract_text_and_images_from_pdf(self):        if not os.path.exists(self.pdf_path):            log.error("执行失败:运行环境没有pdf存储路径")            return         # 创建图像文件夹(如果不存在)        os.makedirs(self.image_folder, exist_ok=True)        log.info("执行成功:图片存储路径读取成功")         # 打开Step 1 pdf_file路径下保存的pdf        with fitz.open(self.pdf_file) as doc:            log.info("执行成功:读取存储路径PDF成功")            # 判断 PDF 是否包含文本内容            has_text = False             extracted_text = ""            extracted_image_text = ""             # 遍历pdf和分页中的文本结果            for i in range(len(doc)):                page = doc.load_page(i)                text = page.get_text()                 # 如果文本存在                if text:                    has_text = True                    # pdf中分页合并输出文本                    extracted_text += f"Page {i + 1}: {text}"                    log.info(f"执行成功:ORC程序成功获取到PDF第(Page {i + 1})页文本信息")                 # 获取所有的图片数量记作i                images = doc.get_page_images(i)                # 遍历所有pdf中的图片,然后将图片保存                for j, img in enumerate(images):                    xref = img[0]                    pix = fitz.Pixmap(doc, xref)  # 从图像创建 pixmap                    img_path = None                    # 处理pdf图片中的核心方法                    if pix.n - pix.alpha < 4:  # 可以直接保存为 PNG                        img_width, img_height = pix.width, pix.height                         # 设置一个阈值,比如小于100像素的图片不保存                        if img_width >= 100 and img_height >= 100:                            img_path = os.path.join(self.image_folder, f"p{i + 1}-{j + 1}.png")                            pix.save(img_path)                            log.info(f"执行成功:ORC程序成功获取到PDF第({i + 1})页获取到(第{j + 1})张PNG图片信息")                        else:                            log.info(                                f"执行成功:ORC程序成功获取到PDF第({i + 1})页获取到(第{j + 1})张PNG图片尺寸较小,不保存")                     else:  # CMYK 图像:必须先转换为 RGB                        pix0 = fitz.Pixmap(fitz.csRGB, pix)                        img_width, img_height = pix.width, pix.height                         # 设置一个阈值,比如小于100像素的图片不保存                        if img_width >= 100 and img_height >= 100:                            img_path = os.path.join(self.image_folder, f"p{i + 1}-{j + 1}.png")                            pix0.save(img_path)                            log.info(f"执行成功:ORC程序成功获取到PDF第({i + 1})页获取到(第{j + 1})张RGB图片信息")                        else:                            log.info(                                f"执行成功:ORC程序成功获取到PDF第({i + 1})页获取到(第{j + 1})张RGB图片尺寸较小,不保存")                        pix0 = None  # 释放 Pixmap 资源                     pix = None  # 释放 Pixmap 资源                     if img_path is not None:                        img_text = pytesseract.image_to_string(Image.open(img_path))                        extracted_image_text += img_text             # 文本值没有获取到时输出log,用来区分order_id和shipping notes是从哪里获取到的文本            if not has_text:                log.info("执行成功:ORC程序没有获取到文本信息")         self.clear_folder()        log.info("执行成功:清空已完成遍历的PDF图片文件")        # 将文本生成的text和图片OCR识别到的text合并为一个        merged_text = extracted_text + extracted_image_text        merged_text = merged_text.replace(" ", "").replace("\n", "")        log.info("执行成功:提取到PDF和图片合并文本信息")        return merged_text


适用场景:

想通过打印面单上面的文字是否存在,或者系统生成的pdf是否包含自己自定义的文字内容。

实现目标:

通过以上代码实现:当没有pdf路径时创建自动创建,再之前的步骤中已在路径下生成了pdf临时存在本地的指定文件夹路径下。

1.然后打开pdf,首先判断pdf中是否存在文本值,如果存在则打上标记。

2.获取pdf中所有的图片个数,然后将其按照 if pix.n - pix.alpha的方式判断是否格式可以存为png。

3.添加图片尺寸验证,防止图片过小。

4.pytesseract.image_to_string将图片转为文字,遍历所有图片将所有的文字合并返回结果。

部分调试:

(图片获取结果)

(图片转为text)

相关推荐

# Python 3 # Python 3字典Dictionary(1)

Python3字典字典是另一种可变容器模型,且可存储任意类型对象。字典的每个键值(key=>value)对用冒号(:)分割,每个对之间用逗号(,)分割,整个字典包括在花括号({})中,格式如...

Python第八课:数据类型中的字典及其函数与方法

Python3字典字典是另一种可变容器模型,且可存储任意类型对象。字典的每个键值...

Python中字典详解(python 中字典)

字典是Python中使用键进行索引的重要数据结构。它们是无序的项序列(键值对),这意味着顺序不被保留。键是不可变的。与列表一样,字典的值可以保存异构数据,即整数、浮点、字符串、NaN、布尔值、列表、数...

Python3.9又更新了:dict内置新功能,正式版十月见面

机器之心报道参与:一鸣、JaminPython3.8的热乎劲还没过去,Python就又双叒叕要更新了。近日,3.9版本的第四个alpha版已经开源。从文档中,我们可以看到官方透露的对dic...

Python3 基本数据类型详解(python三种基本数据类型)

文章来源:加米谷大数据Python中的变量不需要声明。每个变量在使用前都必须赋值,变量赋值以后该变量才会被创建。在Python中,变量就是变量,它没有类型,我们所说的"类型"是变...

一文掌握Python的字典(python字典用法大全)

字典是Python中最强大、最灵活的内置数据结构之一。它们允许存储键值对,从而实现高效的数据检索、操作和组织。本文深入探讨了字典,涵盖了它们的创建、操作和高级用法,以帮助中级Python开发...

超级完整|Python字典详解(python字典的方法或操作)

一、字典概述01字典的格式Python字典是一种可变容器模型,且可存储任意类型对象,如字符串、数字、元组等其他容器模型。字典的每个键值key=>value对用冒号:分割,每个对之间用逗号,...

Python3.9版本新特性:字典合并操作的详细解读

处于测试阶段的Python3.9版本中有一个新特性:我们在使用Python字典时,将能够编写出更可读、更紧凑的代码啦!Python版本你现在使用哪种版本的Python?3.7分?3.5分?还是2.7...

python 自学,字典3(一些例子)(python字典有哪些基本操作)

例子11;如何批量复制字典里的内容2;如何批量修改字典的内容3;如何批量修改字典里某些指定的内容...

Python3.9中的字典合并和更新,几乎影响了所有Python程序员

全文共2837字,预计学习时长9分钟Python3.9正在积极开发,并计划于今年10月发布。2月26日,开发团队发布了alpha4版本。该版本引入了新的合并(|)和更新(|=)运算符,这个新特性几乎...

Python3大字典:《Python3自学速查手册.pdf》限时下载中

最近有人会想了,2022了,想学Python晚不晚,学习python有前途吗?IT行业行业薪资高,发展前景好,是很多求职群里严重的香饽饽,而要进入这个高薪行业,也不是那么轻而易举的,拿信工专业的大学生...

python学习——字典(python字典基本操作)

字典Python的字典数据类型是基于hash散列算法实现的,采用键值对(key:value)的形式,根据key的值计算value的地址,具有非常快的查取和插入速度。但它是无序的,包含的元素个数不限,值...

324页清华教授撰写【Python 3 菜鸟查询手册】火了,小白入门字典

如何入门学习python...

Python3.9中的字典合并和更新,了解一下

全文共2837字,预计学习时长9分钟Python3.9正在积极开发,并计划于今年10月发布。2月26日,开发团队发布了alpha4版本。该版本引入了新的合并(|)和更新(|=)运算符,这个新特性几乎...

python3基础之字典(python中字典的基本操作)

字典和列表一样,也是python内置的一种数据结构。字典的结构如下图:列表用中括号[]把元素包起来,而字典是用大括号{}把元素包起来,只不过字典的每一个元素都包含键和值两部分。键和值是一一对应的...

取消回复欢迎 发表评论:

请填写验证码