Python带你打造个性化的图片文字识别

我们的目标：从CSV文件读取用户的文件信息，并将文件名称修改为姓名格式的中文名称，进行规范资料整理，从而实现快速对多个文件进行重命名。

最终效果：将原来无规律的文件名重命名为以姓名为名称的文件。

技术点：CSV文件读取和保存、os.walk()方法、os.path.splitext()方法、os.path.join()方法、os.rename()方法，类定义、实例方法、实例方法互调、私有方法的封装等。

代码编写方式：采用面向对象方式编写。

接下来我们一起进行代码编写，通过2步搞定这个案例。

（1）ocr接口获取。

首先登陆百度AI开放平台，进入到管理页面后，点击左侧菜单“文字识别”，再点击“创建应用”，如图所示。

创建一个应用，名称为“图片文字识别”，会自动生成AppID和API Key。如图所示。

（2）代码编写。

使用pip命令进行安装，代码如下。

pip install baidu-aip

安装后版本为baidu-aip-2.2.18.0。

首先引入AipOcr类，然后传入AppID,API_Key,Secret_Key 参数，实例化Ocr对象，如下段代码。

client=AipOcr(AppID,API_Key,Secret_Key)

然后调用相关方法进行识别，如下段代码。

result=client.basicGeneral(img) #通用识别，每天5万次免费

result=client.basicAccurate(img) #高精度识别，每天500次免费

通过下段代码演示ocr识别，源代码见code\10\apiocr.py。

from aip import AipOcr

import glob

import os

cur_path=os.path.dirname(__file__)

jpgfile=os.path.join(cur_path, 'ocr',"幻灯片2.jpg")

jpgsfile=os.path.join(cur_path, 'ocr',"*.jpg")

resulttxt=os.path.join(cur_path,'ocr' ,"all.txt")

class ApiOcr():

def __init__(self,appid,api_key,secret_key):

self.appid=appid

self.api_key=api_key

self.secret_key=secret_key

def ocr(self,srcname):

client=AipOcr(self.appid,self.api_key,self.secret_key)

with open(srcname,'rb') as f:

img=f.read()

try:

result=client.basicGeneral(img) #通用识别，每天5万次免费

except:

result=client.basicAccurate(img) #高精度识别，每天500次免费

return result

#批量转化

def ocrs(self,filepath):

for x in glob.glob(filepath):

txts=[]

result=ApiOcr.ocr(self,x)

for txt in result.get('words_result'):

txts.append(txt.get('words'))

print(x)

ApiOcr.__write_file(self,x,txts)

def __write_file(self,picfile,txts):

with open(resulttxt,'a') as f:

f.writelines('识别图片:'+picfile+'\n')

f.writelines('识别内容:'+'\n')

f.writelines(txts)

f.writelines('\n')

if __name__=='__main__':

AppID='21334886'

API_Key='RlMcTlCT0nv7SkRW4wvVcn1T'

Secret_Key='B8SBmqgr2RAVIFBqwaj4WioBOgGEtBBr'

api=ApiOcr(AppID,API_Key,Secret_Key)

print(api.ocr(jpgfile))

print(api.ocrs(jpgsfile))

代码执行结果，对照的图片，生成的文本文件内容如图所示。

大家好，我是【Python办公自动化】：闲暇之余分享点文字、编程、设计等干货，希望和你一起成长。

分享专业全面的Python办公自动化知识，教你快速学习Python的方法，可以站内私信我。