我们的目标:从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识别,源代码见code10apiocr.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+'')
f.writelines('识别内容:'+'')
f.writelines(txts)
f.writelines('')
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的方法,可以站内私信我。