注册百度智能云,选择产品服务。
https://console.bce.baidu.com/
每天赠送200次,做开发测试足够了。
在应用列表复制 AppID , API Key ,Secret Key 备用。
SDK下载地址
https://ai.baidu.com/sdk#ocr
下载SDK文件,解压,复制aip文件夹到项目的根目录下备用。
车牌识别服务费用
运行代码后的效果图
实现代码
# 包的本质上是一个包含了一个或多个模块的文件夹,其中还包含一个特殊的文件 __init__.py
# 要注意引入第三方的包的路径: 包名.模块名 =》 aip.ocr
from aip.ocr import AipOcrfrom PIL import Image, ImageDraw, ImageFont
import numpy as np
import cv2# 修改成自己的
APP_ID = 'xxxxxxxxx'
API_KEY = 'xxxxxxxxxxxxx'
SECRET_KEY = 'xxxxxxxxxxxxxxxxxxxx'
# 创建客户端对象
client = AipOcr(APP_ID, API_KEY, SECRET_KEY)
# 建立连接的超时时间,单位为毫秒
client.setConnectionTimeoutInMillis(5000)
# 通过打开的连接传输数据的超时时间,单位为毫秒
client.setSocketTimeoutInMillis(5000)""" 解决中文乱码 """
def cv2ImgAddText(img, text, left, top, textColor, textSize):# 判断是否OpenCV图片类型if (isinstance(img, np.ndarray)):img = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))# 创建一个可以在给定图像上绘图的对象draw = ImageDraw.Draw(img)# 字体的格式fontStyle = ImageFont.truetype("simsun.ttc", textSize, encoding="utf-8")# 绘制文本draw.text((left, top), text, textColor, font=fontStyle)# 转换回OpenCV格式return cv2.cvtColor(np.asarray(img), cv2.COLOR_RGB2BGR)""" 读取文件内容 """
def getFileContent(filePath):with open(filePath, 'rb') as fp:return fp.read()""" 在图片上画出方框并写上文字 """
def imgDraw(imgPath):image_content = getFileContent(imgPath)# 百度返回的结果res = client.licensePlate(image_content)if res is not None:car_number = res['words_result']['number']car_color = res['words_result']['color']print('车牌号码:' + car_number)print('车牌颜色:' + car_color)# 车牌位置信息location = res['words_result']['vertexes_location']start_x = location[0]['x']start_y = location[0]['y']end_x = location[2]['x']end_y = location[2]['y']# 读取图片img = cv2.imread(imgPath)# 画长方形(图像矩阵,左上角坐标,右下角坐标,颜色,线条粗细)cv2.rectangle(img, (start_x, start_y), (end_x, end_y), (0, 0, 255), 5)# 在图片上绘制文字txt = car_number + ' ' + car_color# cv2ImgAddText(图像, 文字内容, 字体左边开始位置, 字体上面开始位置, (R, G, B), 字体大小)img_txt = cv2ImgAddText(img, txt, start_x+50, start_x+100, (0, 255, 0), 30)# 显示效果cv2.imshow('img', img_txt)# 等待按任意键关闭窗口cv2.waitKey(0)cv2.destroyAllWindows()else:print('车牌识别失败!')if __name__ == '__main__':# 图片路径imgPath = './output/car2.jpg'imgDraw(imgPath)
在代码中要注意引入百度云车牌识别SDK包的路径,否则会报错:ModuleNotFoundError: No module named ‘ocr’ 或者 ImportError: attempted relative import with no known parent package