投稿
  1. 半码博客首页
  2. 后端开发
  3. Python

【PyQt】PyQt+百度API实现图像识别应用（附代码）

小半 • • Python • 阅读 143

导读：本篇文章讲解 【PyQt】PyQt+百度API实现图像识别应用（附代码），希望对大家有帮助，欢迎收藏，转发！站点地址：www.bmabk.com

我是一个甜甜的大橙子🍊，欢迎关注✉️！
我相信技术的力量💪
努力将所学分享给大家😎
你的点赞❤️分享🚀收藏📖就是对我最大的鼓励！

1.效果

选择要识别的类型，然后导入图片，利用百度API，完成图像中的文字识别，将结果显示在右侧，并且可以将识别结果复制到剪贴板。

在这里插入图片描述

2.运行前准备

2.1申请百度AI的API

  1. 这里给出参考文档 AI接入指南，按照步骤进行操作。
  2. 按照步骤操作后会得到API Key和Secret Key，这是获取Access Token的前提。
  3. 注意按照提示，申请免费的测试资源。
  4. 将自己的API Key和Secret Key替换到下面代码中即可。

2.2安装pyqt5等相关依赖包

pip install sip
pip install pyqt5
pip install pyqt5-tools
pip install requests

3代码

在代码中找到AK和SK，替换成2.1中自己的API Key和Secret Key即可。

AK = 'API_KEY'
SK = 'SECRET_KEY'

完整代码

# -*- coding: utf-8 -*-

# Form implementation generated from reading ui file 'face_recognition.ui'
#
# Created by: PyQt5 UI code generator 5.15.4
#
# WARNING: Any manual changes made to this file will be lost when pyuic5 is
# run again.  Do not edit this file unless you know what you are doing.


from PyQt5 import QtCore, QtGui, QtWidgets
from PyQt5.QtCore import *
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
import sys
import requests
import base64
import time


class Ui_Form(object):
    def setupUi(self, Form):
        Form.setObjectName("Form")
        Form.resize(1117, 787)
        self.horizontalLayoutWidget = QtWidgets.QWidget(Form)
        self.horizontalLayoutWidget.setGeometry(QtCore.QRect(130, 120, 421, 80))
        self.horizontalLayoutWidget.setObjectName("horizontalLayoutWidget")
        self.horizontalLayout = QtWidgets.QHBoxLayout(self.horizontalLayoutWidget)
        self.horizontalLayout.setContentsMargins(0, 0, 0, 0)
        self.horizontalLayout.setObjectName("horizontalLayout")
        self.label = QtWidgets.QLabel(self.horizontalLayoutWidget)
        self.label.setObjectName("label")
        self.horizontalLayout.addWidget(self.label)
        self.comboBox = QtWidgets.QComboBox(self.horizontalLayoutWidget)
        self.comboBox.setObjectName("comboBox")
        self.comboBox.addItem("")
        self.comboBox.addItem("")
        self.comboBox.addItem("")
        self.comboBox.addItem("")
        self.comboBox.addItem("")
        self.comboBox.addItem("")
        self.comboBox.addItem("")
        self.comboBox.addItem("")
        self.comboBox.addItem("")
        self.comboBox.addItem("")
        self.comboBox.addItem("")
        self.horizontalLayout.addWidget(self.comboBox)
        self.horizontalLayoutWidget_2 = QtWidgets.QWidget(Form)
        self.horizontalLayoutWidget_2.setGeometry(QtCore.QRect(130, 230, 421, 80))
        self.horizontalLayoutWidget_2.setObjectName("horizontalLayoutWidget_2")
        self.horizontalLayout_2 = QtWidgets.QHBoxLayout(self.horizontalLayoutWidget_2)
        self.horizontalLayout_2.setContentsMargins(0, 0, 0, 0)
        self.horizontalLayout_2.setObjectName("horizontalLayout_2")
        self.label_2 = QtWidgets.QLabel(self.horizontalLayoutWidget_2)
        self.label_2.setObjectName("label_2")
        self.horizontalLayout_2.addWidget(self.label_2)
        self.lineEdit = QtWidgets.QLineEdit(self.horizontalLayoutWidget_2)
        self.lineEdit.setObjectName("lineEdit")
        self.horizontalLayout_2.addWidget(self.lineEdit)
        self.pushButton = QtWidgets.QPushButton(self.horizontalLayoutWidget_2)
        self.pushButton.setObjectName("pushButton")
        self.horizontalLayout_2.addWidget(self.pushButton)
        self.label_3 = QtWidgets.QLabel(Form)
        self.label_3.setGeometry(QtCore.QRect(130, 340, 421, 311))
        self.label_3.setStyleSheet("border-width:1px; border-style:solid; border-color:rgb(0,0,0)")
        self.label_3.setText("")
        self.label_3.setObjectName("label_3")
        self.verticalLayoutWidget = QtWidgets.QWidget(Form)
        self.verticalLayoutWidget.setGeometry(QtCore.QRect(609, 119, 321, 531))
        self.verticalLayoutWidget.setObjectName("verticalLayoutWidget")
        self.verticalLayout = QtWidgets.QVBoxLayout(self.verticalLayoutWidget)
        self.verticalLayout.setContentsMargins(0, 0, 0, 0)
        self.verticalLayout.setObjectName("verticalLayout")
        self.label_4 = QtWidgets.QLabel(self.verticalLayoutWidget)
        self.label_4.setStyleSheet("border-width:1px; border-style:solid; border-color:rgb(0,0,0)")
        self.label_4.setText("")
        self.label_4.setObjectName("label_4")
        self.verticalLayout.addWidget(self.label_4)
        self.pushButton_2 = QtWidgets.QPushButton(self.verticalLayoutWidget)
        self.pushButton_2.setObjectName("pushButton_2")
        self.verticalLayout.addWidget(self.pushButton_2)

        self.retranslateUi(Form)
        QtCore.QMetaObject.connectSlotsByName(Form)

    def retranslateUi(self, Form):
        _translate = QtCore.QCoreApplication.translate
        Form.setWindowTitle(_translate("Form", "图像识别工具"))
        self.label.setText(_translate("Form", "选择识别类型"))
        self.comboBox.setItemText(0, _translate("Form", "银行卡"))
        self.comboBox.setItemText(1, _translate("Form", "动物"))
        self.comboBox.setItemText(2, _translate("Form", "植物"))
        self.comboBox.setItemText(3, _translate("Form", "通用票据"))
        self.comboBox.setItemText(4, _translate("Form", "营业执照"))
        self.comboBox.setItemText(5, _translate("Form", "身份证"))
        self.comboBox.setItemText(6, _translate("Form", "车牌号"))
        self.comboBox.setItemText(7, _translate("Form", "驾驶证"))
        self.comboBox.setItemText(8, _translate("Form", "行驶证"))
        self.comboBox.setItemText(9, _translate("Form", "车型"))
        self.comboBox.setItemText(10, _translate("Form", "Logo"))
        self.label_2.setText(_translate("Form", "选择图片"))
        self.pushButton.setText(_translate("Form", "选择"))
        self.pushButton.clicked.connect(self.openfile)
        self.pushButton_2.setText(_translate("Form", "复制到剪贴板"))
        self.pushButton_2.clicked.connect(self.copyResult)

    def openfile(self):
        self.file_path = QtWidgets.QFileDialog.getOpenFileName(self.horizontalLayoutWidget_2, "选择要识别的图片", '/',
                                                               '*.jpg *.png')
        if not self.file_path[0].strip():
            pass
        else:
            self.lineEdit.setText(self.file_path[0])
            pixmap = QtGui.QPixmap(self.file_path[0])
            scaled_pixmap = pixmap.scaled(QtCore.QSize(311, 300), aspectRatioMode=Qt.KeepAspectRatio)
            self.label_3.setPixmap(scaled_pixmap)
            self.type_selected()

    def copyResult(self):
        clipboard = QApplication.clipboard()
        clipboard.setText(self.label_4.text())
        self.child_window = Child()
        self.child_window.show()

    def type_selected(self):
        '''
        0.银行卡
        1.动物
        2.植物
        3.通用票据
        4.营业执照
        5.身份证
        6.车牌号
        7.驾驶证
        8.行驶证
        9.车型
        10.Logo
        '''
        if self.comboBox.currentIndex() == 0:
            pass
        if self.comboBox.currentIndex() == 1:
            pass
        if self.comboBox.currentIndex() == 2:
            pass
        if self.comboBox.currentIndex() == 3:
            pass
        if self.comboBox.currentIndex() == 4:
            pass
        if self.comboBox.currentIndex() == 5:
            self.get_idcard(self.get_token())
        if self.comboBox.currentIndex() == 6:
            pass
        if self.comboBox.currentIndex() == 7:
            pass
        if self.comboBox.currentIndex() == 8:
            pass
        if self.comboBox.currentIndex() == 9:
            pass
        if self.comboBox.currentIndex() == 10:
            pass

    def get_token(self):
        # client_id 为官网获取的AK， client_secret 为官网获取的SK
        AK = 'API_KEY'
        SK = 'SECRET_KEY'
        host = 'https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=' + AK + '&client_secret=' + SK
        response = requests.get(host)
        access_token = response.json().get('access_token')
        return access_token

    def get_idcard(self, access_token):
        '''
        身份证识别
        '''
        request_url = "https://aip.baidubce.com/rest/2.0/ocr/v1/idcard"
        # 二进制方式打开图片文件
        f = open(self.file_path[0], 'rb')
        img = base64.b64encode(f.read())

        params = {"id_card_side": "front", "image": img}
        access_token = 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)
        info_idcard = response.json()
        try:
            words_result = info_idcard['words_result']
            result_recoginition = '识别结果为： \n'
            result_recoginition += '姓名：{} \n性别：{} \n民族：{} \n出生日期：{} \n住址：{} \n公民身份证号码：{}\n'.format(
                words_result['姓名']['words'],
                words_result['性别']['words'],
                words_result['民族']['words'],
                words_result['出生']['words'],
                words_result['住址']['words'],
                words_result['公民身份号码']['words']
            )
        except BaseException:
            result_recoginition = '无法解析'
        self.label_4.setText(result_recoginition)


class Child(QWidget):
    def __init__(self):
        super().__init__()
        self.setWindowTitle('复制成功')
        layout = QVBoxLayout()
        self.label = QLabel('复制成功')
        layout.addWidget(self.label)
        self.setLayout(layout)


if __name__ == '__main__':
    app = QtWidgets.QApplication(sys.argv)
    MainWindow = QtWidgets.QMainWindow()
    ui = Ui_Form()
    ui.setupUi(MainWindow)
    MainWindow.show()
    sys.exit(app.exec_())

版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 举报，一经查实，本站将立刻删除。

文章由半码博客整理，本文链接：https://www.bmabk.com/index.php/post/63033.html

赞 (0)
小半的头像小半
0 0

相关推荐

半码博客——专业性很强的中文编程技术网站，欢迎收藏到浏览器，订阅我们！