图像识别API Python 指定文件夹图片内容转化为文字
程序员文章站
2024-03-16 10:40:16
...
有个室友被无良老师叫去把课本上的东西打到电脑上。
于是就有了以下的东西 -Xerath
效果展示:
原图
翻译结果
代码:
# -*- coding: UTF-8 -*-
import os
from aip import AipOcr
import json
APP_ID = '11030410'
API_KEY = 'nFL4SLu5GjpM2K1aGQZPKidO'
SECRET_KEY = 'QZpXrgt6XLFcC8IfzVQViWcORLVHzc28'
aipOcr = AipOcr(APP_ID, API_KEY, SECRET_KEY)
# 指定文件夹(拿去用的同学只要改这里)
os.chdir("g:\\yangzai")
dirs = os.listdir()
def get_file_content(filePath):
with open(filePath, 'rb') as fp:
return fp.read()
options = {
'detect_direction': 'true',
'language_type': 'CHN_ENG',
}
print('开始处理,共'+str(len(dirs))+"张图片。")
cnt=0
for filePath in dirs:
if filePath.split('.')[-1]=='txt':continue
cnt+=1
print('正在处理第'+str(cnt)+'张图片')
result = aipOcr.basicGeneral(get_file_content(filePath), options)
with open(filePath.split('.')[0]+'.txt','w',encoding='utf-8') as ans:
for i in result['words_result']:
ans.write(i['words']+'\n')
print('处理完成')
print('全部处理完成!')