Python实例之调用百度API实现车牌识别

Posted 2021-07-11 ZHW_AI课题组

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了Python实例之调用百度API实现车牌识别相关的知识，希望对你有一定的参考价值。

作者介绍

王世豪，男，西安工程大学电子信息学院，2020级硕士研究生，张宏伟人工智能课题组。
研究方向：机器视觉与人工智能。
电子邮件：shauwang@foxmail.com

API介绍

支持识别中国大陆机动车蓝牌、黄牌（单双行）、绿牌、大型新能源（黄绿）、领使馆车牌、警牌、武警牌（单双行）、军牌（单双行）、港澳出入境车牌、农用车牌、民航车牌的地域编号和车牌号，并能同时识别图像中的多张车牌。

调用过程

首先打开百度智能云官网注册登录。
在这里插入图片描述
登录之后选择文字识别服务

创建应用

然后输入应用名称、描述，并选择应用类型为个人，之后点击“立即创建”按钮。

之后返回应用列表就可以看到已经创建好的车牌识别的应用，此处显示API Key和Secret Key，后边程序中会用到。
在这里插入图片描述

注意：百度智能云车牌识别API免费版提供每天200次免费调用，后续会调用失败。建议妥善保存API Key和Secret Key。

应用创建完成之后打开接口文档https://ai.baidu.com/ai-doc/OCR/ck3h7y191得到以下有用信息。
请求URL: https://aip.baidubce.com/rest/2.0/ocr/v1/license_plate
Header格式：Content-Type：application/x-www-form-urlencoded
在这里插入图片描述

代码实现

使用示例代码前，请记得替换其中的示例图片地址和Key。

# -*- coding: utf-8 -*-
import urllib
import urllib.parse
import urllib.request
import base64
import json

# client_id 为官网获取的API Key， client_secret 为官网获取的Secret Key，将自己应用的Key分别复制在下方即可。
client_id = '*********'
client_secret = '********'

# 获取token
def get_token():
    host = 'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=' + client_id + '&client_secret=' + client_secret
    request = urllib.request.Request(host)
    request.add_header('Content-Type', 'application/json; charset=UTF-8')
    response = urllib.request.urlopen(request)
    token_content = response.read()
    if token_content:
        token_info = json.loads(token_content.decode("utf-8"))
        token_key = token_info['access_token']
    return token_key

# 读取图片
def get_file_content(filePath):
    with open(filePath, 'rb') as fp:
        return fp.read()
        
# 获取车牌号信息
def get_license_plate(path):
    request_url = "https://aip.baidubce.com/rest/2.0/ocr/v1/license_plate"

    f = get_file_content(path)
    access_token = get_token()
    img = base64.b64encode(f)
    params = {"custom_lib": False, "image": img}
    params = urllib.parse.urlencode(params).encode('utf-8')
    request_url = request_url + "?access_token=" + access_token
    request = urllib.request.Request(url=request_url, data=params)
    request.add_header('Content-Type', 'application/x-www-form-urlencoded')
    response = urllib.request.urlopen(request)
    content = response.read()
    # print("json类型下的返回值：", content)
    if content:
        license_plates = json.loads(content.decode("utf-8"))
        # print("utf-8解码并转化为python类型下的返回值：", license_plates)
        strover = '识别结果：'
        words_result = license_plates['words_result']
        # print("python类型下的返回值：", words_result)
        number = words_result['number']
        strover += '  车牌号：{} \\n '.format(number)
        print(strover)
        return content
    else:
        return ''
# 读取图片，自行保存图片，更改路径即可。
image_path = 'D:\\桌面\\chepai.jpg'
# image_path = 'C:/Users/Wong/chepai.jpg'
# image_path = r'C:\\Users\\Wong\\chepai.jpg'
get_license_plate(image_path)