import glob from os import path import os from aip import AipOcr from PIL import Image def convertimg(picfile, outdir): ‘‘‘调整图片大小,对于过大的图片进行压缩 picfile: 图片路径 outdir: 图片输出路径 ‘‘‘ img = Image.open(picfile) width, height = img.size while(width*height > 4000000): # 该数值压缩后的图片大约 两百多k width = width // 2 height = height // 2 new_img=img.resize((width,height),Image.BILINEAR) #合并目录path.join(...)。你给几个目录它就合并几个,形参多个目录间以逗号隔开 new_img.save(path.join(outdir,os.path.basename(picfile))) print("我是convertimg函数下的:"+os.path.basename(picfile)) def baiduOCR(picfile, outfile): """利用百度api识别文本,并保存提取的文字 picfile: 图片文件名 outfile: 输出文件 """ filename = path.basename(picfile) APP_ID = "" # 百度文字识别的 ID,需要注册账号后获取,下同 API_KEY = "" SECRECT_KEY = "" client = AipOcr(APP_ID, API_KEY, SECRECT_KEY) i = open(picfile, ‘rb‘) img = i.read() print("正在识别图片:\t" + filename) message = client.basicGeneral(img) # 通用文字识别,每天 50 000 次免费 #message = client.basicAccurate(img) # 通用文字高精度识别,每天 800 次免费 print("识别成功!") i.close(); with open(outfile, ‘a+‘) as fo: fo.writelines("+" * 60 + ‘\n‘) fo.writelines("识别图片:\t" + filename + "\n" * 2) fo.writelines("文本内容:\n") # 输出文本内容 for text in message.get(‘words_result‘): fo.writelines(text.get(‘words‘) + ‘\n‘) fo.writelines(‘\n‘*2) print("文本导出成功!") print() if __name__ == "__main__": outfile = r‘H:/picture/export.txt‘ outdir = r‘H:/picture/tmp‘ if path.exists(outfile): os.remove(outfile) if not path.exists(outdir): print("创建临时输出目录: "+outdir) os.mkdir(outdir) print("压缩过大的图片...") #首先对过大的图片进行压缩,以提高识别速度,将压缩的图片保存于临时文件夹中 for picfile in glob.glob("H:/picture/*.jpg"): convertimg(picfile, outdir) print("图片识别...") for picfile in glob.glob("H:/picture/tmp/*"): baiduOCR(picfile, outfile) os.remove(picfile) print(‘图片文本提取结束!文本输出结果位于 %s 文件中。‘ % outfile) os.removedirs(outdir)
原文地址:https://www.cnblogs.com/caihongmin/p/10934110.html
时间: 2024-10-08 22:42:39