devbox
很楠不爱3
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

用Python编写百度ocr图像识别程序_利用百度ocr python

作者：很楠不爱3 | 2024-05-31 14:37:47

赞

踩

利用百度ocr python

涉及的知识点：Python、调用百度API、PyQt5

运行环境：Mac+Python3.7+PyQt5

程序截图：

实现步骤：

第一步：获得自己的百度OCR的AK和SK

1、利用百度 AI 开发平台的 OCR 文字识别 API 识别并提取图片中的文字，首先需注册获取 API 调用的 ID 和 key，打开百度AI开放平台，进入控制台中的文字识别应用(需要有百度账号）。 

2、创建一个应用，并进入管理应用，记下 AppID, API Key, Secrect Key，调用 API需用到。 

 

第二步：PyQt5安装和pycharm配置

1.直接使用pip安装,但是需要SIP的支持,所以先安装SIP,再安装pyqt5

pip install sip 
pip install PyQt5

2.安装qt designer，同样使用pip安装

pip install PyQt5-tools

3.配置PyCharm，配置PyCharm是为了在Pycharm里面实现打开qt designer，生成qt文件，方便转换成python文件。

具体步骤可移步查看这篇文章：pyqt5安装与pycharm配置

 

第三步：用qt designer编写界面

新建一个项目，然后在左边窗口部件盒找到需要的控件，拖动到界面编辑区，保存baidu_ocr.ui

 

第四步：对刚才保存的baidu_ocr.ui文件右键选择 External Tools-pyUIC 功能，将QT文件转化为py文件

第五步：编辑相关代码，运行之前请将AK和SK替换为自己的

  1. # -*- coding: utf-8 -*-
  2.  
  3. '''
  4. author: 神奇的老黄
  5. gtihub: https://github.com/MH-Blog
  6. site: https://magichuang.cn
  7. date: 2019.02.14 23:08
  8. '''
  9.  
  10. import base64
  11. import json
  12.  
  13. from PyQt5 import QtCore, QtGui, QtWidgets
  14.  
  15. import sys
  16. import urllib, urllib.request
  17. import ssl
  18.  
  19. from PyQt5.QtCore import QSize, Qt
  20.  
  21. ssl._create_default_https_context = ssl._create_unverified_context
  22.  
  23. """ 你的 APPID AK SK """
  24. # http://ai.baidu.com/docs#/OCR-Pricing/top 申请地址
  25.  
  26. # API_KEY 为官网获取的AK， SECRET_KEY 为官网获取的SK
  27. API_KEY = 'API_KEY'
  28. SECRET_KEY = 'SECRET_KEY'
  29.  
  30.  
  31. # QT自动生成的UI类
  32. class Ui_Form(object):
  33.     def setupUi(self, Form):
  34.         # 设置窗体的名称
  35.         Form.setObjectName("Form")
  36.         # 设置窗体大小
  37.         Form.resize(834, 789)
  38.         # 新建横向布局
  39.         self.horizontalLayoutWidget = QtWidgets.QWidget(Form)
  40.         self.horizontalLayoutWidget.setGeometry(QtCore.QRect(40, 30, 251, 61))
  41.         self.horizontalLayoutWidget.setObjectName("horizontalLayoutWidget")
  42.         self.horizontalLayout = QtWidgets.QHBoxLayout(self.horizontalLayoutWidget)
  43.         self.horizontalLayout.setContentsMargins(0, 0, 0, 0)
  44.         self.horizontalLayout.setObjectName("horizontalLayout")
  45.         self.label = QtWidgets.QLabel(self.horizontalLayoutWidget)
  46.         self.label.setObjectName("label")
  47.         self.horizontalLayout.addWidget(self.label)
  48.         self.comboBox = QtWidgets.QComboBox(self.horizontalLayoutWidget)
  49.         # 设置名称
  50.         self.comboBox.setObjectName("comboBox")
  51.         # 添加子标题有多少分类添加几个
  52.         self.comboBox.addItem("")
  53.         self.comboBox.addItem("")
  54.         self.comboBox.addItem("")
  55.         self.comboBox.addItem("")
  56.         self.comboBox.addItem("")
  57.         self.comboBox.addItem("")
  58.         self.comboBox.addItem("")
  59.         self.comboBox.addItem("")
  60.         self.comboBox.addItem("")
  61.         self.comboBox.addItem("")
  62.         self.comboBox.addItem("")
  63.         # 把下拉选择控件添加到布局容器中
  64.         self.horizontalLayout.addWidget(self.comboBox)
  65.         # 新建横向布局
  66.         self.horizontalLayoutWidget_2 = QtWidgets.QWidget(Form)
  67.         self.horizontalLayoutWidget_2.setGeometry(QtCore.QRect(350, 30, 441, 61))
  68.         self.horizontalLayoutWidget_2.setObjectName("horizontalLayoutWidget_2")
  69.         self.horizontalLayout_2 = QtWidgets.QHBoxLayout(self.horizontalLayoutWidget_2)
  70.         self.horizontalLayout_2.setContentsMargins(0, 0, 0, 0)
  71.         self.horizontalLayout_2.setObjectName("horizontalLayout_2")
  72.         self.label_2 = QtWidgets.QLabel(self.horizontalLayoutWidget_2)
  73.         self.label_2.setObjectName("label_2")
  74.         self.horizontalLayout_2.addWidget(self.label_2)
  75.         self.lineEdit = QtWidgets.QLineEdit(self.horizontalLayoutWidget_2)
  76.         self.lineEdit.setObjectName("lineEdit")
  77.         self.horizontalLayout_2.addWidget(self.lineEdit)
  78.         # 新建按钮
  79.         self.pushButton = QtWidgets.QPushButton(self.horizontalLayoutWidget_2)
  80.         self.pushButton.setObjectName("pushButton")
  81.         self.horizontalLayout_2.addWidget(self.pushButton)
  82.         self.label_3 = QtWidgets.QLabel(Form)
  83.         self.label_3.setGeometry(QtCore.QRect(40, 110, 751, 301))
  84.         self.label_3.setAutoFillBackground(False)
  85.         # 设置黑色边框
  86.         self.label_3.setStyleSheet('border-width:1px;border-style:solid;border-color:rgb(0,0,0);')
  87.         self.label_3.setObjectName("label_3")
  88.         self.horizontalLayoutWidget_3 = QtWidgets.QWidget(Form)
  89.         self.horizontalLayoutWidget_3.setGeometry(QtCore.QRect(40, 440, 751, 321))
  90.         self.horizontalLayoutWidget_3.setObjectName("horizontalLayoutWidget_3")
  91.         self.horizontalLayout_3 = QtWidgets.QHBoxLayout(self.horizontalLayoutWidget_3)
  92.         self.horizontalLayout_3.setContentsMargins(0, 0, 0, 0)
  93.         self.horizontalLayout_3.setObjectName("horizontalLayout_3")
  94.         self.verticalLayout_2 = QtWidgets.QVBoxLayout()
  95.         self.verticalLayout_2.setObjectName("verticalLayout_2")
  96.         self.label_4 = QtWidgets.QLabel(self.horizontalLayoutWidget_3)
  97.         self.label_4.setAutoFillBackground(False)
  98.         # self.label_4.setStyleSheet("background-color: rgb(253, 188, 49);")
  99.         self.label_4.setObjectName("label_4")
  100.         self.label_4.setStyleSheet('border-width:1px;border-style:solid;border-color:rgb(0,0,0);')
  101.         self.verticalLayout_2.addWidget(self.label_4)
  102.         # 新建按钮
  103.         self.pushButton_2 = QtWidgets.QPushButton(self.horizontalLayoutWidget_3)
  104.         self.pushButton_2.setObjectName("pushButton_2")
  105.         self.verticalLayout_2.addWidget(self.pushButton_2, 0, QtCore.Qt.AlignHCenter)
  106.         self.horizontalLayout_3.addLayout(self.verticalLayout_2)
  107.         # 自动生成的ui控件 处理方法
  108.         self.retranslateUi(Form)
  109.         # 自动生成的  关联信号槽
  110.         QtCore.QMetaObject.connectSlotsByName(Form)
  111.  
  112.     def retranslateUi(self, Form):
  113.         _translate = QtCore.QCoreApplication.translate
  114.         # 设置窗体内容
  115.         Form.setWindowTitle(_translate("Form", "百度图片识别程序"))
  116.         # 设置文字控件显示内容
  117.         self.label.setText(_translate("Form", "识别类型："))
  118.         # 设置下拉控件选项内容
  119.         self.comboBox.setItemText(0, _translate("Form", "银行卡"))
  120.         self.comboBox.setItemText(1, _translate("Form", "植物"))
  121.         self.comboBox.setItemText(2, _translate("Form", "动物"))
  122.         self.comboBox.setItemText(3, _translate("Form", "通用票据"))
  123.         self.comboBox.setItemText(4, _translate("Form", "营业执照"))
  124.         self.comboBox.setItemText(5, _translate("Form", "身份证"))
  125.         self.comboBox.setItemText(6, _translate("Form", "车牌号"))
  126.         self.comboBox.setItemText(7, _translate("Form", "驾驶证"))
  127.         self.comboBox.setItemText(8, _translate("Form", "行驶证"))
  128.         self.comboBox.setItemText(9, _translate("Form", "车型"))
  129.         self.comboBox.setItemText(10, _translate("Form", "Logo"))
  130.  
  131.         self.label_2.setText(_translate("Form", "图片路径："))
  132.         self.pushButton.setText(_translate("Form", "选择"))
  133.         self.label_3.setText(_translate("Form", "<html><head/><body><p align=\"center\">此处显示照片</p></body></html>"))
  134.         self.label_4.setText(_translate("Form", "<html><head/><body><p align=\"center\">此处显示结果</p></body></html>"))
  135.         self.pushButton_2.setText(_translate("Form", "复制到剪贴板"))
  136.  
  137.         # 为 选择图片 按钮添加点击事件
  138.         self.pushButton.clicked.connect(self.openFile)
  139.         # 为 复制到剪贴板 按钮添加点击时间
  140.         self.pushButton_2.clicked.connect(self.copyText)
  141.  
  142.     #  复制到剪贴板
  143.     def copyText(self):
  144.         clipboard = QtWidgets.QApplication.clipboard()
  145.         clipboard.setText(self.label_4.text())
  146.         # QMessageBox.about(self,'关于','这是一个关于消息对话框!')
  147.         msgBox = QtWidgets.QMessageBox(QtWidgets.QMessageBox.NoIcon, '提醒', '识别内容已成功复制到剪贴板 !')
  148.         msgBox.exec()
  149.  
  150.     # 打开文件选择对话框方法
  151.     def openFile(self):
  152.         # 启动选择文件对话框，查找jpg以及png图片
  153.         self.download_path = QtWidgets.QFileDialog.getOpenFileName(self.horizontalLayoutWidget, '选择要识别的图片', '',
  154.                                                                    'Image File (*.jpg *.jpeg *.png)')
  155.         # 判断是否选择图片
  156.         if not self.download_path[0].strip():
  157.             pass
  158.         else:
  159.             # 将路径显示到地址栏中
  160.             self.lineEdit.setText(self.download_path[0])
  161.             # 将图片显示到图片展示区
  162.             pixmap = QtGui.QPixmap(self.download_path[0])
  163.             # self.label_3.setPixmap(pixmap)
  164.             # 等比例缩放图片
  165.             scaredPixmap = pixmap.scaled(QSize(311, 301), aspectRatioMode=Qt.KeepAspectRatio)
  166.             # 设置图片
  167.             self.label_3.setPixmap(scaredPixmap)
  168.             # 判断选择的类型 根据类型做相应的图片处理
  169.             self.label_3.show()
  170.             # 判断选择的类型
  171.             self.typeTp()
  172.  
  173.  
  174.     # 判断选择的类型 进行相应处理
  175.     def typeTp(self):
  176.         # 银行卡识别
  177.         if self.comboBox.currentIndex() == 0:
  178.             self.get_bankcard(self.get_token())
  179.             pass
  180.         # 植物识别
  181.         elif self.comboBox.currentIndex() == 1:
  182.             self.get_plant(self.get_token())
  183.             pass
  184.         # 动物识别
  185.         elif self.comboBox.currentIndex() == 2:
  186.             self.get_animal(self.get_token())
  187.             pass
  188.         # 通用票据识别识别
  189.         elif self.comboBox.currentIndex() == 3:
  190.             self.get_vat_invoice(self.get_token())
  191.             pass
  192.         # 营业执照识别
  193.         elif self.comboBox.currentIndex() == 4:
  194.             self.get_business_licensev(self.get_token())
  195.             pass
  196.         # 身份证识别
  197.         elif self.comboBox.currentIndex() == 5:
  198.             self.get_idcard(self.get_token())
  199.             pass
  200.         # 车牌号识别
  201.         elif self.comboBox.currentIndex() == 6:
  202.             self.get_license_plate(self.get_token())
  203.             pass
  204.         # 驾驶证识别
  205.         elif self.comboBox.currentIndex() == 7:
  206.             self.get_driving_license(self.get_token())
  207.             pass
  208.         # 行驶证识别
  209.         elif self.comboBox.currentIndex() == 8:
  210.             self.get_vehicle_license(self.get_token())
  211.             pass
  212.         # 车型识别
  213.         elif self.comboBox.currentIndex() == 9:
  214.             self.get_car(self.get_token())
  215.             pass
  216.         # Logo识别
  217.         elif self.comboBox.currentIndex() == 10:
  218.             self.get_logo(self.get_token())
  219.             pass
  220.         pass
  221.  
  222.     # 获取token
  223.     def get_token(self):
  224.         # client_id 为官网获取的AK， client_secret 为官网获取的SK
  225.         host = 'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=' + API_KEY + '&client_secret=' + SECRET_KEY
  226.         # 发送请求
  227.         request = urllib.request.Request(host)
  228.         # 添加请求头
  229.         request.add_header('Content-Type', 'application/json; charset=UTF-8')
  230.         # 获取返回内容
  231.         response = urllib.request.urlopen(request)
  232.         # 读取返回内容
  233.         content = response.read()
  234.         # 判断内容是否为空
  235.         if (content):
  236.             # 打印内容
  237.             print(content)
  238.             # 打印token
  239.             print(json.loads(content)['access_token'])
  240.             # 使用json解析出token 设置token
  241.             self.access_token = json.loads(content)['access_token']
  242.             # 返回token
  243.             return self.access_token
  244.  
  245.     # 0 银行卡识别
  246.     def get_bankcard(self, access_token):
  247.         request_url = "https://aip.baidubce.com/rest/2.0/ocr/v1/bankcard"
  248.         # 二进制方式打开图片文件
  249.         f = self.get_file_content(self.download_path[0])
  250.         img = base64.b64encode(f)
  251.         params = {"image": img}
  252.         params = urllib.parse.urlencode(params).encode('utf-8')
  253.         request_url = request_url + "?access_token=" + access_token
  254.         request = urllib.request.Request(url=request_url, data=params)
  255.         request.add_header('Content-Type', 'application/x-www-form-urlencoded')
  256.         response = urllib.request.urlopen(request)
  257.         content = response.read()
  258.         if content:
  259.             # 解析返回数据
  260.             bankcards = json.loads(content)
  261.             # 输出返回结果
  262.             strover = '识别结果：\n'
  263.             # 捕捉异常判断是否正确返回信息
  264.             try:
  265.                 # 判断银行卡类型
  266.                 if bankcards['result']['bank_card_type'] == 0:
  267.                     bank_card_type = '不能识别'
  268.                 elif bankcards['result']['bank_card_type'] == 1:
  269.                     bank_card_type = '借记卡'
  270.                 elif bankcards['result']['bank_card_type'] == 2:
  271.                     bank_card_type = '信用卡'
  272.                 strover += '  卡号：{} \n  银行：{} \n  类型：{} \n'.format(bankcards['result']['bank_card_number'],
  273.                                                                    bankcards['result']['bank_name'], bank_card_type)
  274.             # 错误的时候提示错误原因
  275.             except BaseException:
  276.                 error_msg = bankcards['error_msg']
  277.                 strover += '  错误：\n {} \n '.format(error_msg)
  278.             # 设置识别显示结果
  279.             self.label_4.setText(strover)
  280.  
  281.     # 1 植物识别
  282.     def get_plant(self, access_token):
  283.         request_url = "https://aip.baidubce.com/rest/2.0/image-classify/v1/plant"
  284.         # 二进制方式打开图片文件
  285.         f = self.get_file_content(self.download_path[0])
  286.         # 转换图片
  287.         img = base64.b64encode(f)
  288.         # 拼接图片参数
  289.         params = {"image": img}
  290.         params = urllib.parse.urlencode(params).encode('utf-8')
  291.         # 请求地址
  292.         request_url = request_url + "?access_token=" + access_token
  293.         # 发送请求传递图片参数
  294.         request = urllib.request.Request(url=request_url, data=params)
  295.         # 添加访问头部
  296.         request.add_header('Content-Type', 'application/x-www-form-urlencoded')
  297.         # 接收返回内容
  298.         response = urllib.request.urlopen(request)
  299.         # 读取返回内容
  300.         content = response.read()
  301.         # 内容判断
  302.         if content:
  303.             plants = json.loads(content)
  304.             strover = '识别结果：\n'
  305.             try:
  306.                 i = 1
  307.                 for plant in plants['result']:
  308.                     strover += '{} 植物名称：{} \n'.format(i, plant['name'])
  309.                     i += 1
  310.             except BaseException:
  311.                 error_msg = plants['error_msg']
  312.                 strover += '  错误：\n {} \n '.format(error_msg)
  313.             self.label_4.setText(strover)
  314.  
  315.     # 2 动物识别
  316.     def get_animal(self, access_token):
  317.         request_url = "https://aip.baidubce.com/rest/2.0/image-classify/v1/animal"
  318.         # 二进制方式打开图片文件
  319.         f = self.get_file_content(self.download_path[0])
  320.         img = base64.b64encode(f)
  321.         params = {"image": img, "top_num": 6}
  322.         params = urllib.parse.urlencode(params).encode('utf-8')
  323.         request_url = request_url + "?access_token=" + access_token
  324.         request = urllib.request.Request(url=request_url, data=params)
  325.         request.add_header('Content-Type', 'application/x-www-form-urlencoded')
  326.         response = urllib.request.urlopen(request)
  327.         content = response.read()
  328.         if content:
  329.             animals = json.loads(content)
  330.             strover = '识别结果：\n'
  331.             try:
  332.                 i = 1
  333.                 for animal in animals['result']:
  334.                     strover += '{} 动物名称：{} \n'.format(i, animal['name'])
  335.                     i += 1
  336.             except BaseException:
  337.                 error_msg = animals['error_msg']
  338.                 strover += '  错误：\n {} \n '.format(error_msg)
  339.             self.label_4.setText(strover)
  340.  
  341.     # 3 通用票据识别
  342.     def get_vat_invoice(self, access_token):
  343.         request_url = "https://aip.baidubce.com/rest/2.0/ocr/v1/receipt"
  344.         # 二进制方式打开图片文件
  345.         f = self.get_file_content(self.download_path[0])
  346.         img = base64.b64encode(f)
  347.         params = {"image": img}
  348.         params = urllib.parse.urlencode(params).encode('utf-8')
  349.         request_url = request_url + "?access_token=" + access_token
  350.         request = urllib.request.Request(url=request_url, data=params)
  351.         request.add_header('Content-Type', 'application/x-www-form-urlencoded')
  352.         response = urllib.request.urlopen(request)
  353.         content = response.read()
  354.         if content:
  355.             receipts = json.loads(content)
  356.             strover = '识别结果：\n'
  357.             try:
  358.                 words_result = receipts['words_result']
  359.                 for word_result in words_result:
  360.                     # 票据内容
  361.                     InvoiceType = word_result['words']
  362.                     strover += ' {} '.format(InvoiceType)
  363.             except BaseException:
  364.                 error_msg = receipts['error_msg']
  365.                 strover += '  错误：\n {} \n '.format(error_msg)
  366.             self.label_4.setText(strover)
  367.  
  368.     # 4 营业执照识别
  369.     def get_business_licensev(self, access_token):
  370.         request_url = "https://aip.baidubce.com/rest/2.0/ocr/v1/business_license"
  371.         # 二进制方式打开图片文件
  372.         f = self.get_file_content(self.download_path[0])
  373.         img = base64.b64encode(f)
  374.         params = {"image": img}
  375.         params = urllib.parse.urlencode(params).encode('utf-8')
  376.         request_url = request_url + "?access_token=" + access_token
  377.         request = urllib.request.Request(url=request_url, data=params)
  378.         request.add_header('Content-Type', 'application/x-www-form-urlencoded')
  379.         response = urllib.request.urlopen(request)
  380.         content = response.read()
  381.         if content:
  382.             business_licenses = json.loads(content)
  383.             strover = '识别结果：\n'
  384.             try:
  385.                 words_result = business_licenses['words_result']
  386.                 # 单位名称
  387.                 Unit_name = words_result['单位名称']['words']
  388.                 strover += '  单位名称：\n  {} \n '.format(Unit_name)
  389.                 # 法人
  390.                 legal_person = words_result['法人']['words']
  391.                 strover += '  法人：{} \n '.format(legal_person)
  392.                 # 有效期
  393.                 Term_of_validity = words_result['有效期']['words']
  394.                 strover += '  有效期：{} \n '.format(Term_of_validity)
  395.                 # 证件编号
  396.                 ID_number = words_result['证件编号']['words']
  397.                 strover += '  证件编号：{} \n '.format(ID_number)
  398.                 # 社会信用代码
  399.                 Social_Credit_Code = words_result['社会信用代码']['words']
  400.                 strover += '  社会信用代码：{} \n '.format(Social_Credit_Code)
  401.                 # 地址
  402.                 address = words_result['地址']['words']
  403.                 strover += '  地址：\n{}\n '.format(address)
  404.             except BaseException:
  405.                 error_msg = business_licenses['error_msg']
  406.                 strover += '  错误：\n  {} \n '.format(error_msg)
  407.             self.label_4.setText(strover)
  408.  
  409.     # 5 身份证识别
  410.     def get_idcard(self, access_token):
  411.         request_url = "https://aip.baidubce.com/rest/2.0/ocr/v1/idcard"
  412.         # 二进制方式打开图片文件
  413.         f = self.get_file_content(self.download_path[0])
  414.         img = base64.b64encode(f)
  415.         params = {"image": img, "id_card_side": "front"}
  416.         params = urllib.parse.urlencode(params).encode('utf-8')
  417.         request_url = request_url + "?access_token=" + access_token
  418.         request = urllib.request.Request(url=request_url, data=params)
  419.         request.add_header('Content-Type', 'application/x-www-form-urlencoded')
  420.         response = urllib.request.urlopen(request)
  421.         content = response.read()
  422.         if content:
  423.             idcards = json.loads(content)
  424.             strover = '识别结果：\n'
  425.             try:
  426.                 words_result = idcards['words_result']
  427.                 # 公民身份号码
  428.                 Citizenship_number = words_result['公民身份号码']['words']
  429.                 strover += '  公民身份号码：\n{} \n '.format(Citizenship_number)
  430.                 # 民族
  431.                 Nation = words_result['民族']['words']
  432.                 strover += '  民族：{} \n '.format(Nation)
  433.                 # 姓名
  434.                 Full_name = words_result['姓名']['words']
  435.                 strover += '  姓名：{} \n '.format(Full_name)
  436.                 # 住址
  437.                 address = words_result['住址']['words']
  438.                 strover += '  住址：\n{} \n '.format(address)
  439.             except BaseException:
  440.                 error_msg = idcards['error_msg']
  441.                 strover += '  错误：\n  {} \n '.format(error_msg)
  442.             # 显示识别结果
  443.             self.label_4.setText(strover)
  444.  
  445.     # 6 车牌号识别
  446.     def get_license_plate(self, access_token):
  447.         request_url = "https://aip.baidubce.com/rest/2.0/ocr/v1/license_plate"
  448.         # 二进制方式打开图片文件
  449.         f = self.get_file_content(self.download_path[0])
  450.         img = base64.b64encode(f)
  451.         params = {"image": img}
  452.         params = urllib.parse.urlencode(params).encode('utf-8')
  453.         request_url = request_url + "?access_token=" + access_token
  454.         request = urllib.request.Request(url=request_url, data=params)
  455.         request.add_header('Content-Type', 'application/x-www-form-urlencoded')
  456.         response = urllib.request.urlopen(request)
  457.         content = response.read()
  458.         if content:
  459.             license_plates = json.loads(content)
  460.             strover = '识别结果：\n'
  461.             try:
  462.                 words_result = license_plates['words_result']
  463.                 # 车牌号
  464.                 number = words_result['number']
  465.                 strover += '  车牌号：{} \n '.format(number)
  466.             except BaseException:
  467.                 error_msg = license_plates['error_msg']
  468.                 strover += '  错误：\n  {} \n '.format(error_msg)
  469.             self.label_4.setText(strover)
  470.  
  471.     # 7 驾驶证识别
  472.     def get_driving_license(self, access_token):
  473.         request_url = "https://aip.baidubce.com/rest/2.0/ocr/v1/driving_license"
  474.         # 二进制方式打开图片文件
  475.         f = self.get_file_content(self.download_path[0])
  476.         img = base64.b64encode(f)
  477.         params = {"image": img}
  478.         params = urllib.parse.urlencode(params).encode('utf-8')
  479.         request_url = request_url + "?access_token=" + access_token
  480.         request = urllib.request.Request(url=request_url, data=params)
  481.         request.add_header('Content-Type', 'application/x-www-form-urlencoded')
  482.         response = urllib.request.urlopen(request)
  483.         content = response.read()
  484.         if content:
  485.             driving_licenses = json.loads(content)
  486.             strover = '识别结果：\n'
  487.             try:
  488.                 words_result = driving_licenses['words_result']
  489.                 # 证号
  490.                 Citizenship_number = words_result['证号']['words']
  491.                 strover += '  证号： {} \n '.format(Citizenship_number)
  492.                 # 准驾车型
  493.                 Full_name = words_result['准驾车型']['words']
  494.                 strover += ' 准驾车型：{} \n '.format(Full_name)
  495.                 # 姓名
  496.                 name = words_result['姓名']['words']
  497.                 strover += ' 姓名： {} \n '.format(name)
  498.                 # 国籍
  499.                 nationality = words_result['国籍']['words']
  500.                 strover += ' 国籍： {} \n '.format(nationality)
  501.                 # 出生日期
  502.                 date_of_birth = words_result['出生日期']['words']
  503.                 strover += ' 出生日期： {} \n '.format(date_of_birth)
  504.                 # 性别
  505.                 sex = words_result['性别']['words']
  506.                 strover += ' 性别： {} \n '.format(sex)
  507.                 # 初次领证日期
  508.                 first_certificate_date = words_result['初次领证日期']['words']
  509.                 strover += ' 初次领证日期： {} \n '.format(first_certificate_date)
  510.                 # 有效期限
  511.                 Nation = words_result['有效期限']['words']
  512.                 # 到期日期
  513.                 to = words_result['至']['words']
  514.                 strover += ' 有效期限：{}至{}\n '.format(Nation, to)
  515.                 # 住址
  516.                 address = words_result['住址']['words']
  517.                 strover += ' 住址：\n{} \n '.format(address)
  518.             except BaseException:
  519.                 error_msg = driving_licenses['error_msg']
  520.                 strover += '  错误：\n  {} \n '.format(error_msg)
  521.             self.label_4.setText(strover)
  522.  
  523.     # 8 行驶证识别
  524.     def get_vehicle_license(self, access_token):
  525.         request_url = "https://aip.baidubce.com/rest/2.0/ocr/v1/vehicle_license"
  526.         # 二进制方式打开图片文件
  527.         f = self.get_file_content(self.download_path[0])
  528.         img = base64.b64encode(f)
  529.         params = {"image": img}
  530.         params = urllib.parse.urlencode(params).encode('utf-8')
  531.         request_url = request_url + "?access_token=" + access_token
  532.         request = urllib.request.Request(url=request_url, data=params)
  533.         request.add_header('Content-Type', 'application/x-www-form-urlencoded')
  534.         response = urllib.request.urlopen(request)
  535.         content = response.read()
  536.         if content:
  537.             vehicle_licenses = json.loads(content)
  538.             strover = '识别结果：\n'
  539.             try:
  540.                 words_result = vehicle_licenses['words_result']
  541.                 # 品牌型号
  542.                 brand_model = words_result['品牌型号']['words']
  543.                 strover += '  品牌型号： {} \n '.format(brand_model)
  544.                 # 发证日期
  545.                 date_of_certification = words_result['发证日期']['words']
  546.                 strover += ' 发证日期：{} \n '.format(date_of_certification)
  547.                 # 使用性质
  548.                 use_nature = words_result['使用性质']['words']
  549.                 strover += ' 使用性质： {} \n '.format(use_nature)
  550.                 # 发动机号码
  551.                 engine_number = words_result['发动机号码']['words']
  552.                 strover += ' 发动机号码： {} \n '.format(engine_number)
  553.                 # 注册日期
  554.                 date_of_registration = words_result['注册日期']['words']
  555.                 strover += ' 注册日期： {} \n '.format(date_of_registration)
  556.                 # 号牌号码
  557.                 number_number = words_result['号牌号码']['words']
  558.                 strover += ' 号牌号码： {} \n '.format(number_number)
  559.                 # 车辆识别代号
  560.                 vehicle_identification = words_result['车辆识别代号']['words']
  561.                 strover += ' 车辆识别代号： {} \n '.format(vehicle_identification)
  562.                 # 车辆类型
  563.                 vehicle_type = words_result['车辆类型']['words']
  564.                 strover += ' 车辆类型： {} \n '.format(vehicle_type)
  565.                 # 所有人
  566.                 owner = words_result['所有人']['words']
  567.                 strover += ' 所有人：\n{} \n '.format(owner)
  568.                 # 住址
  569.                 address = words_result['住址']['words']
  570.                 strover += ' 住址：\n{} \n '.format(address)
  571.             except BaseException:
  572.                 error_msg = vehicle_licenses['error_msg']
  573.                 strover += '  错误：\n  {} \n '.format(error_msg)
  574.             self.label_4.setText(strover)
  575.  
  576.     # 9 获取车型信息
  577.     def get_car(self, access_token):
  578.         request_url = "https://aip.baidubce.com/rest/2.0/image-classify/v1/car"
  579.         # 二进制方式打开图片文件
  580.         f = self.get_file_content(self.download_path[0])
  581.         img = base64.b64encode(f)
  582.         params = {"image": img, "top_num": 5}
  583.         params = urllib.parse.urlencode(params).encode('utf-8')
  584.         request_url = request_url + "?access_token=" + access_token
  585.         request = urllib.request.Request(url=request_url, data=params)
  586.         request.add_header('Content-Type', 'application/x-www-form-urlencoded')
  587.         response = urllib.request.urlopen(request)
  588.         content = response.read()
  589.         if content:
  590.             cars = json.loads(content)
  591.             strover = '识别结果：\n'
  592.             try:
  593.                 i = 1
  594.                 for car in cars['result']:
  595.                     strover += '{} 车型：{} \n  年份：{} \n'.format(i, car['name'], car['year'])
  596.                     i += 1
  597.             except BaseException:
  598.                 error_msg = cars['error_msg']
  599.                 strover += '  错误：\n  {} \n '.format(error_msg)
  600.             self.label_4.setText(strover)
  601.  
  602.     # 10 获取logo信息
  603.     def get_logo(self, access_token):
  604.         request_url = "https://aip.baidubce.com/rest/2.0/image-classify/v2/logo"
  605.         # 二进制方式打开图片文件
  606.         f = self.get_file_content(self.download_path[0])
  607.         img = base64.b64encode(f)
  608.         params = {"custom_lib": False, "image": img}
  609.         params = urllib.parse.urlencode(params).encode('utf-8')
  610.         request_url = request_url + "?access_token=" + access_token
  611.         request = urllib.request.Request(url=request_url, data=params)
  612.         request.add_header('Content-Type', 'application/x-www-form-urlencoded')
  613.         response = urllib.request.urlopen(request)
  614.         content = response.read()
  615.         if content:
  616.             logos = json.loads(content)
  617.             strover = '识别结果：\n'
  618.             try:
  619.                 i = 1
  620.                 for logo in logos['result']:
  621.                     strover += '{} Logo名称：{} \n'.format(i, logo['name'])
  622.                     i += 1
  623.             except BaseException:
  624.                 error_msg = logos['error_msg']
  625.                 strover += '  错误：\n  {} \n '.format(error_msg)
  626.             self.label_4.setText(strover)
  627.  
  628.     # 读取图片
  629.     def get_file_content(self, filePath):
  630.         with open(filePath, 'rb') as fp:
  631.             return fp.read()
  632.  
  633.  
  634. if __name__ == '__main__':
  635.     app = QtWidgets.QApplication(sys.argv)
  636.     MainWindow = QtWidgets.QMainWindow()
  637.     ui = Ui_Form()
  638.     ui.setupUi(MainWindow)
  639.     MainWindow.show()
  640.     sys.exit(app.exec_())

为了方便测试，已经将所有项目文件和所需的测试图片打包上传了

下载地址：百度图像自动识别

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/很楠不爱3/article/detail/652848
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号