Python调用百度AI实现图片上文字识别功能实例

简介

Python免费调用百度AI实现图片上面的文字识别

步骤

安装百度AI库

!pip install baidu-aip

注册百度AI开放平台

先注册百度AI,获得ID和密钥。注册方法可参考:注册方法 只需走到 “1.6 获取密钥” 即可。然后记录下自己的APP_ID、API_KEY、SECRET_KEY,就可以开始了。

调用glob库

glob库用于获得指定路径下的指定后缀的文件,图片使用的是《数学模型》pdf扫描版,部分如下:

Python调用百度AI实现图片上文字识别功能实例_第1张图片

import glob
path = "数学模型\\"
glob.glob(path+"*.png")

指定文件夹中的指定的.png后缀的文件,全部提取出来了:

选取第一张图片 “数学模型1.png” 做测试:

files = glob.glob(path+"*.png")
file = files[0]

调用AipOcr库识别文字

导入AipOcr模块,这个模块是用于做文字识别的(OCR即Optical Character Recognition,光学字符识别)。填入你申请的ID和KEY,client = AipOcr()是固定写法。先用二进制方式rb打开,然后读取,再调用百度AI进行识别,并将识别结果存入message变量。

from aip import AipOcr #导入AipOcr模块,用于做文字识别
import glob

APP_ID = '*********' # 你申请的
API_KEY = '*********'# 你申请的
SECRET_KEY = '*********'# 你申请的
client = AipOcr(APP_ID, API_KEY, SECRET_KEY)

pic = open(file,'rb')# 以二进制(rb)打开
img = pic.read() # 读取
message = client.basicGeneral(img) # 调用百度AI识别图片中的文字

查看message参数,结果如下:

Python调用百度AI实现图片上文字识别功能实例_第2张图片

可见,message参数的结果为 “大字典” 形式,其中的键 words_result 所对应的值为一个 “大列表” ,这个列表里又有很多键值对,键都是words,值就是AI从图片上识别出来的内容。使用语句message['words_result'][0...n]['words']可以筛选出该页的标题:

message['words_result'][1]['words']

for循环逐行提取:

for words in message['words_result']:
    word = words['words'] # 提取文字
    print(word)

识别效果图:

Python调用百度AI实现图片上文字识别功能实例_第3张图片

识别也不是完美的,有很多识别错误与未识别出的文字。当然,原图字体越清晰,识别度越高。

可能会遇到的问题

识别过程中可能会遇到类似识别错误(无法识别)问题,原因是未领取免费识别次数,解决过程如下:

百度OCR API识别失败:Open api qps request limit reached 错误 解决方法

几个月前,用Python写了个小工具,主要是用到了百度的OCR API做了个文字识别的小工具,前面一直能用,然而今天,在使用的时候,报错了。
错误如下:Open api qps request limit reached

最后终于找到了解决方法,原来,现在要自己去领取免费的测试额度了!!!

领取免费测试额度的地址:(或者你在你原来那个调用的那个项目里也可以找到免费领取测试额度,见下图)

https://console.bce.baidu.com/ai/#/ai/ocr/overview/resource/getFree

领取完之后,还是用原来项目的appID,apiKey,secretKey即可。

然后等领取额度到账后,重新运行就行了。

Python调用百度AI实现图片上文字识别功能实例_第4张图片

批量操作

批量操作的核心为for循环,txt_file = open('数学模型.txt', 'a')指打开名字为“数学模型”的文本文件(若不存在则自动新建),模式为添加模式,即'a'模式。添加模式是指,如果文本中已经有一些内容,则接着往下写,不会影响原来的内容。如果是'w'模式,则会将之前的内容全部覆盖。然后遍历待处理的图片,逐个处理:

from aip import AipOcr #导入AipOcr模块,用于做文字识别
import glob

APP_ID = '*********' # 你申请的
API_KEY = '*********'# 你申请的
SECRET_KEY = '*********'# 你申请的
client = AipOcr(APP_ID, API_KEY, SECRET_KEY)

path = "数学模型\\"
files = glob.glob(path+"*.png")

txt_file = open('数学模型.txt', 'a')

for file in files:
    pic = open(file,'rb')# 以二进制(rb)打开
    img = pic.read() # 读取
    message = client.basicGeneral(img) # 调用百度AI识别图片中的文字
    for words in message['words_result']:
        word = words['words'] # 提取文字
        print(word)

        txt_file.write(word) # 将文字写入文本文件

txt_file.close() # 关闭文本文件

总结

到此这篇关于Python调用百度AI实现图片上文字识别功能的文章就介绍到这了,更多相关Python图片上文字识别内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

你可能感兴趣的