百度AI图像识别

import requests
import base64
import os

AK = 'xxx'
SK = 'xxx'

# client_id 为官网获取的AK， client_secret 为官网获取的SK
host = f'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id={AK}&client_secret={SK}'
response = requests.get(host)
if response:
    print(response.json())


def get_id_card_info(path):
    print(path)
    # client_id 为官网获取的AK， client_secret 为官网获取的SK
    host = f'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id={AK}&client_secret={SK}'
    # host = f'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id={AK}&client_secret={SK}'
    response = requests.get(host)
    if response:
        taken = response.json()
        # print(taken)
        # print(taken['access_token'])
        '''
        
        '''
        request_url = " https://aip.baidubce.com/rest/2.0/ocr/v1/accurate_basic"  # 多卡证类别检测
        # request_url = "https://aip.baidubce.com/rest/2.0/ocr/v1/idcard" #身份证识别
        f = open(path, 'rb')
        img = base64.b64encode(f.read())
        params = {"image": img}
        access_token = taken['access_token']
        request_url = request_url + "?access_token=" + access_token
        headers = {'content-type': 'application/x-www-form-urlencoded'}
        response = requests.post(request_url, data=params, headers=headers)
        if response:
            res = response.json()
            print(res)

            # print(res)
            # with open('id_card_log.txt', 'a') as f:
            #     # f = open('log.txt', 'a')
            #     f.write(f'{name}-{id}！\n')
            #     f.close()
            #     print(f'{name}-{id}-信息提取成功！')


# 提取身份证照片目录
jpgs_ls = os.listdir('测试/')
for jpg_name in jpgs_ls:
    path = '测试/' + jpg_name
    get_id_card_info(path)

data_ls = []
for i in d['words_result']:
    ls = []
    if '姓名' in i['words']:
        ls.append(i['words'])
    if '身份证号' in i['words']:
        ls.append(i['words'])
    if '学号' in i['words']:
        ls.append(i['words'])
    if ls:
        data_ls.append(ls)
print(data_ls)