first commit

This commit is contained in:
张乾
2024-10-16 06:37:41 +08:00
parent 633f45ea20
commit 206fad82a2
3590 changed files with 680090 additions and 0 deletions

View File

@@ -0,0 +1,251 @@
因收到Google相关通知网站将会择期关闭。相关通知内容
03 图片转文字:如何提高识别准确率?
你好,我是尹会生。
不知道你有没有遇见过这样的场景:在工作中,你遇见了一个紧急情况,对方给你发了一串儿聊天记录的长截图,当你处理完事情想要复盘的时候,必须要把这些记录处理成文字,然后就发现图片转文字过程中会出现很多问题,还需要自己二次手动校对。
经过不断尝试就发现用互联网上的AI产品可以非常准确地识别出图片中的印刷体文字。而且再通过Python还可以实现把识别到的文字进行格式处理、文件保存等自动化操作。
那么今天,我就给你介绍两种能够精准、快速地把图片转成文字的方式:在线识别和离线识别。我会给你讲解具体的操作方法,当你再遇见这样的需求的时候,就可以很轻松应对了。
图片转文字的两种处理方法
我先来对图片转文字的两种处理方法进行介绍。
目前能够达到较高文字识别正确率的一般分为两种识别方式:一种是文字识别工作都需要在网络侧完成的方式,我们称为在线识别;另一种是不需要互联网功能的,我们称作离线识别。
根据不同的工作场景,我会选择不同的方式实现文字识别。那么,接下来我就带你了解下这两种方式各自的特性。
先看第一种,在线识别的方式。
在线识别方式最大的优点就是它在初次进行文字识别的时候准确率非常高。比如对聊天截图中的识别准确率就高达99%。因为在线识别使用了人工智能领域的深度学习算法和文字识别相结合的技术,能够把图片转换成文字后,还能在语义上把相近的字进行二次纠正。
比如说被识别的内容包含英文单词“Hello”一旦它的字母“o”被识别成数字“0”在线识别软件就会根据上下文语境把这类错误纠正回来而这种二次纠正的功能在离线识别软件中是没有的。
不过在线识别软件也有它的缺点那就是识别文字的过程需要在公有云的服务器上完成。也就是说需要通过互联网把图片上传到服务器那么一旦图片过大或者图片数量比较多就会导致上传时间过长。我们知道一张高清图片至少有3MB大小根据个人的网络情况至少要达到秒级上传才行。这就意味着在大批量文字识别的场景中或对实时性要求很高的场景下在线识别是不能满足要求的。
另外,图片需要经过互联网传输,识别以后的图片该怎么保存,怎么销毁,是不是会被其他人得到,这些都是安全风险。总之,信息泄露的风险比较大。所以像公司的合同、财务资料等涉密程度比较高的扫描件,很少使用在线识别。
再看第二种,离线识别的方式。
这种方式在识别过程中不需要连接网络,节省了在线传输图片的时间,适合那些对实时性要求比较高或网络信号比较差的场景。
但是离线识别方式的问题就在于,初次识别文字的准确率比较低,识别完之后必须要经过人工二次纠正才行。所以在前期人工校对,花费的时间相对来说会比较长。
我把它们各自的优缺点做了一张表格,如下:
那接下来,我就带你学习一下这两种识别方式怎样具体实现。
怎么进行在线文字识别?
我们先来看怎么进行在线文字识别。
在线文字识别方式,识别的主要功能的需要放在公有云的服务器中才能实现,所以在代码实现中就要考虑用户验证和图片加密传输问题。
用户验证能确保识别的结果交还给你本人图片加密传输能确保图片上的信息不会被其他人窃取到。这些功能各个公有云的AI产品都考虑得非常周全一般会提供给用户一个扩展库。你要做的就是安装这些扩展库。
举个例子百度云的AI产品你可以在终端下执行这样一个命令来进行安装。
pip install baidu-aip
在这里我使用了百度云提供的在线文字识别产品它提供了一个baidu-aip的安装包安装之后提供了AipOcr函数实现用户验证、client.basicGeneral函数实现文字识别功能。代码如下
from aip import AipOcr
""" 你的 APPID AK SK """
# APP_ID = '你的 App ID'
# API_KEY = '你的 Api Key'
# SECRET_KEY = '你的 Secret Key'
client = AipOcr(APP_ID, API_KEY, SECRET_KEY)
""" 读取图片 """
def get_file_content(filePath):
with open(filePath, 'rb') as fp:
return fp.read()
image = get_file_content('example.png')
""" 调用通用文字识别, 图片参数为本地图片 """
result = client.basicGeneral(image)
print(result)
在这段代码里实现了三个功能分别是用户验证、读取图片和识别图片。为了让你更直观地看到效果我来对一张电子发票的图片example.png进行识别。识别结果如下
其实识别的准确率是非常高的,基本上没有错误。那接下来我具体给你介绍一下核心代码。
在代码的第一行我使用了一个AipOcr库。AipOcr是百度云提供给用户的OCR Python SDK客户端能够让你用Python语言和百度云进行交互。
一般情况下我们进行用户认证、图片上传至服务器功能都需要自己编写很多代码但是使用AipOcr库之后这些基础功能都被封装好了。你只需要填写三个变量就能正式进入文字识别的环节了
第一个变量是APP_ID它用来识别应用
第二个变量是API_KEY用于识别用户
第三个变量是SECRET_KEY ,用来加密密钥。
当把这三个变量传入AipOcr函数使用AipOcr函数通过互联网交互后就可以用来识别用户是不是被授权使用相应的产品之后就可以把图片加密发送到AI产品的服务器上了。
这些就是所有公有云识别用户的通用做法,我把这个做法整理为三个步骤。
第一步, 安装SDK。代码是
pip install baidu-aip
第二步,注册用户。
以百度云为例你需要登录https://ai.baidu.com/网址,以自己的手机为用户名注册一个新的用户。
第三步,申请应用。
成功登录网站之后你会进入服务控制台界面然后选择文字识别功能再新创建一个文字识别类型的应用。创建应用之后就可以在服务控制台中的应用列表中查看百度云提供的APP_ID、API_KEY、SECRET_KEY 三个变量。
在这三个变量中APP_ID在百度云的控制台中创建用户之后会自动创建。-
变量API_KEY与SECRET_KEY是在创建完毕应用后分配给用户的。
在你成功通过服务控制台得到这三个变量之后就可以把这三个变量回添到代码中的第3-5行然后去掉“#”开头的注释,这样就可以通过代码完成用户签名和验证功能。
你在掌握它们之后,也可以使用其他的公有云完成类似的用户签名和验证功能。
完成以上三个步骤之后我们需要把指定路径的图片上传到百度云通过第12行的get_file_content函数把图片的路径和名称作为参数传入这个函数之后再交给client.basicGeneral(image)函数处理这样就能够完成图片的上传功能了图片的识别和返回结果都会由AipOcr包自动处理以后放入result变量中。
总结来说,我在代码中实现了单个图片的在线识别文字功能,当你掌握了对单个图片实现文字识别的方法之后, 你就可以通过我们在第二讲学过的的for循环功能遍历多个文件夹里的多个图片你可以实现批量图片的识别功能了。
怎么进行离线文字识别?
我们再来看另一种文字识别形式——离线文字识别。被识别的图片同样是example.png这张发票的扫描件代码如下
import pytesseract
from PIL import Image
# 打开图片
image = Image.open('example.png')
# 转为灰度图片
imgry = image.convert('L')
# 二值化采用阈值分割算法threshold为分割点,根据图片质量调节
threshold = 150
table = []
for j in range(256):
if j < threshold:
table.append(0)
else:
table.append(1)
temp = imgry.point(table, '1')
# OCR识别lang指定中文,--psm 6 表示按行识别,有助于提升识别准确率
text = pytesseract.image_to_string(temp, lang="chi_sim+eng", config='--psm 6')
# 打印识别后的文本
print(text)
在进行离线识别的时候有一个现成的文字识别库那就是pytesseract库这个库实现了对图片中的文字识别功能使用pytesseract库可以自动实现文字的切分和识别功能识别效果如下
通过识别结果你可以发现英文和数字被准确识别出来的正确率是比较高的达到90%以上如果是纯英文的图片识别之后的文字是可以直接使用的但是在对中文字母数字混合的图片识别结果中中文会出现较多的错误正确率不足50%这种情况下就需要人工去对每个字进行一遍纠正
不过如果我们想让初次识别的正确率就很高那能不能实现呢其实是可以的不过在这个过程中我们需要花大量的时间把自己工作场景中经常出现的字手工进行纠错形成新的识别模型实现起来还是有一定难度的为了让你更好地评估是不是值得自己优化离线识别软件的识别正确率我先来带你看一下原理
对图像进行文字识别的一般过程为5个步骤-
第一步图像输入-
第二步前期处理比如二值化图像降噪倾斜纠正-
第三步文字检测比如版面分析字符分割-
第四步文本识别比如字符识别后期矫正-
第五步也就是最后一步输出文本
在这5个步骤中影响识别准确率最大的就是文本识别部分这部分会受到对文字的标注和算法模型的大小这两个关键因素的影响既然我们搞清楚了图像识别的步骤和原理那能不能在保留本地识别的优势前提下通过在线识别产品来优化离线识别产品的正确率呢我们来具体分析一下
首先我们可以在文字标注方面优化识别正确率
像身份证识别票据识别聊天截图等大部分文字识别场景这些待识别图片采用了印刷体每个字的间隔都是固定的所以我们要想提升准确率通常的做法是对每个字都进行人工纠错
当Tesseract再识别到同一个字的时候就会按照我们纠正过的来识别那么逐个汉字纠正是离线识别方式提升识别正确率的必要方式所以已经有高手为你写好了汉字纠正的工具了它就是 jTessBoxEditorFX软件
具体来说这个软件可以把图片中的每个字切割开并提示哪个方框对应哪个识别的文字更正为正确的文字之后就可以把更正的结果进行保存这个保存好的结果就被称作新训练的模型
所以整个过程你可以理解成三个步骤-
第一步人工观察-
第二步对比原始图像-
第三步把错误的文字手工纠正为正确的汉字
比如截图中的多次识别错误当我们人工把这些文字改为正确的字之后就可以生成一个训练文件再把训练文件和文字识别的算法合并成新的识别模型等下次再识别发票的时候就可以采用新的模型进行识别所以再次出现就可以正确识别了
这种文字标注对于单一场景非常有效比如名片火车票飞机票发票车牌的识别场景经过手工标注不断增加样本数量之后能够让正确率呈对数级别增加
接下来我们再看一下在算法的模型上能不能通过在线识别软件来优化离线识别软件
在线识别软件的模型很大通常是上百GB的容量离线识别软件的模型往往只有几十MB它们之间存在着巨大的差异模型越大样本数量就越多类似离线识别软件手工标注的模型一样巨大的模型让在线识别软件的准确率在第一次识别就会非常高
如果把在线模型直接放到你的笔记本当中因为模型庞大本地计算能力不足识别的时间甚至超过在线识别的时间那本地识别就不具有实时性的优势了所以直接使用在线识别软件的模型方式不适合离线识别软件
总的来说由于手工标注和模型大小有限两个因素的影响离线识别软件的应用场景大部分在企业内部票据识别上面虽然离线识别中文的效果比较差但是如果场景单一在线识别的实时性达不到你的要求那还是值得投入一部分时间人工标注汉字的
识别出文字以后我们就要提取文字内容并保存了
识别成功后的文字处理工作
通过刚才案例中的识别结果我们知道识别结果准确率很高像单张图片已经可以把识别的结果直接保存使用了不过一旦需要识别的图片比较多手工操作起来就非常繁琐了因为识别过程是使用Python编写的程序后续可以继续利用Python对识别结果进行处理和保存比如说在线识别的结果
为了能够提取文本内容去掉记录分段信息的word “{ }”,需要对这段文字再加工处理
如果你仔细观察的话就会发现这段文字是有一定规律的它的写法就是Python的基本数据类型字典”,字典类型会强调一对或多对数据之间的映射关系为了将识别的文字结果进行保存我们还会使用另一个基本数据类型列表”,一般用来表示多段文字的并列关系为了提取文本内容我接下来把字典转换成列表就可以实现内容的提取了
从字典到列表转换的功能怎么实现呢首先来看一下两种数据类型的定义
字典由花扩号中的KEY和VALUE两部分组成的它的格式是{“words:1234567 }
列表由方括号的VALUE组成可以存放一个也可以存放多个VALUE格式是[“1234567”,] [“123”, abc”]。
观察这两个数据类型不难发现把字典的VALUE存放到列表里就能实现内容的提取了如果字典里有多个值可以使用for循环进行遍历然后批量提取对于识别的结果我们可以采用一段代码进行从字典到列表的转换代码如下
info = []
for i in result['words_result']:
info.append(i['words'])
print(info)
经过数据类型的转换以后我们可以得到处理好的文字内容
最后我们把处理好的结果保存至文件就行了因为类型转换和保存我们都使用Python完成所以我可以把这部分功能和文字识别功能放在同一个脚本中避免手工操作
不过我还要提醒你一下在实现用户认证图片上传文字识别数据清洗文件保存的过程中我们要尽可能实现流程的全自动化避免中间步骤手工介入这样可以更高效地提升工作效率
小结
这节课我给你提供了两种文字识别的方式在线识别和离线识别不管哪一种识别方式它们都需要经过图片输入图片预处理文字检测文字识别和输出这样5个部分来实现完整的文字的识别功能
但因为实现方式不同两种方式各有优缺点在线方式准确率更高离线方式保密性更强你需要根据自己的工作场景来选择不同的识别方式不过不管哪种识别最后我都建议你用Python对识别后的结果实现数据清洗和保存工作毕竟减少手工操作才是实现办公效率提升最有效方式
现在你可以开始动手将自己电脑里存储多年的图片自动转成文字了
思考题
在讲离线文字识别的时候我只讲了单张图片转换文字的操作方法如果是多张图片呢该怎么实现呢我建议你使用第二节课学过的for循环来进行操作
欢迎把你对今天这节课的思考和想法分享在留言区我们一起交流讨论也欢迎你点击请朋友读”,我们一起提高工作效率
编辑小提示专栏的完整代码位置是https://github.com/wilsonyin123/python_productivity可点击链接下载查看或者通过网盘链接提取后下载链接是: https://pan.baidu.com/s/1UvEKDCGnU6yb0a7gHLSE4Q?pwd=5wf1提取码: 5wf1