百度文字识别OCR模板

使用须知

python 代码复制到本地 py 文件

img 文件夹放到与 py 文件同级目录

设置好自己申请的 API_KeySecret_Key

申请地址:https://ai.baidu.com/tech/ocr

下面有演示用的程序,直接套就好了

Python代码

import os
import requests
import base64
from urllib.parse import urlencode

API_Key = 'Qw26WGp5wPoqYnIkMePYgcTG'
Secret_Key = 'FyhvKaKN8HxnLeNCssSgoKzV1sYGekBc'
images_path = r'C:\Users\hyzsj\Desktop\img'

if not API_Key or not Secret_Key:
    print('API_Key 或 Secret_Key 不能为空')
    exit(0)

# 获取必须的密钥
def get_token():
    base_url = 'https://aip.baidubce.com/oauth/2.0/token?'
    params = {
        'grant_type': 'client_credentials',
        'client_id': API_Key,
        'client_secret': Secret_Key
    }
    token_url = base_url + urlencode(params)
    response = requests.get(token_url)
    return response.json().get('access_token')


# 发送图片到接口,返回结果
def post_img(img):
    '''
    通用文字识别高精度版
    :return:
    '''
    # 文字识别高精度版,其他请访问:https://ai.baidu.com/ai-doc/OCR/1k3h7y3db
    base_url = "https://aip.baidubce.com/rest/2.0/ocr/v1/accurate_basic"
    data = {"image": img}
    # get_token() = '[调用鉴权接口获取的token]'
    request_url = base_url + "?access_token=" + get_token()
    headers = {'Content-Type': 'application/x-www-form-urlencoded'}
    response = requests.post(request_url, data=data, headers=headers)
    if response:
        return response.json()

# 遍历文件夹中的图片,调用识别api
def find_img():
    for img in os.listdir(images_path):
        # 二进制方式打开图片文件
        f = open(f'{images_path}\{img}', 'rb')
        img = base64.b64encode(f.read())
        f.close()
        text_list.append(post_img(img))


# 结果写入文件
def write_text(words_result: list):
    path = os.path.dirname(os.path.abspath(__file__))
    with open(f'{path}\\result.txt', 'a+', encoding='utf-8')as f:
        for _ in words_result:
            f.write(_.get('words') + '\n')


if __name__ == "__main__":
    text_list=[]
    find_img()
    for _ in text_list:
        words_result = _.get('words_result')
        write_text(words_result)

 

版权声明:
作者:hyzsj0106
链接:https://www.cxywy.com/?p=349
来源:程序员无垠
文章版权归作者所有,未经允许请勿转载。

THE END
分享
二维码
打赏
< <上一篇
下一篇>>