欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

图像识别API Python 指定文件夹图片内容转化为文字

程序员文章站 2024-03-16 10:40:16
...

有个室友被无良老师叫去把课本上的东西打到电脑上。
于是就有了以下的东西 -Xerath

效果展示:
图像识别API Python 指定文件夹图片内容转化为文字
原图
图像识别API Python 指定文件夹图片内容转化为文字
翻译结果
图像识别API Python 指定文件夹图片内容转化为文字
代码:

# -*- 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('全部处理完成!')
相关标签: python 图像识别