import sys
from PyQt5 import QtWidgets, QtCore
from PyQt5.QtCore import Qt
from PyQt5.QtGui import QPixmap, QPalette, QBrush
from PyQt5.QtWidgets import QFileDialog
from PyQt5.uic.properties import QtGui
from aip import AipImageClassify
class Ui_imageAI(object):
def setupUi(self, imageAI):
imageAI.setObjectName("imageAI")
imageAI.resize(724, 489) # 设置窗体大小
#imageAI.setStyleSheet("#imageAI{border-image:url(D:\py_project\\a.png);}")
# 图片显示控件
self.image = QtWidgets.QLabel(imageAI)
self.image.setGeometry(QtCore.QRect(96, 140, 311, 301))
self.image.setStyleSheet("border-width: 1px;border-style: solid;border-color: rgb(0, 0, 0);")
self.widget = QtWidgets.QWidget(imageAI)
self.widget.setGeometry(QtCore.QRect(110, 50, 221, 31))
self.widget.setObjectName("widget")
# 选择识别类型
self.label = QtWidgets.QLabel(self.widget)
self.label.setObjectName("label")
# 下拉控件，用于选择识别类型
self.comboBox = QtWidgets.QComboBox(self.widget)
self.comboBox.setObjectName("comboBox")
self.comboBox.addItem("")
self.comboBox.addItem("")
self.comboBox.addItem("")
# 横向布局容器，包含 self.label self.comboBox
self.HBoxLayout = QtWidgets.QHBoxLayout(self.widget)
self.HBoxLayout.setContentsMargins(0, 0, 0, 0)
self.HBoxLayout.setObjectName("HBoxLayout")
self.HBoxLayout.addWidget(self.label)
self.HBoxLayout.addWidget(self.comboBox)
self.widget2 = QtWidgets.QWidget(imageAI)
self.widget2.setGeometry(QtCore.QRect(96, 90, 318, 31))
self.widget2.setObjectName("widget2")
# 选择要识别的图片
self.label2 = QtWidgets.QLabel(self.widget2)
self.label2.setObjectName("label2")
# 图片路径显示
self.lineEdit = QtWidgets.QLineEdit(self.widget2)
self.lineEdit.setObjectName("lineEdit")
# 图片路径选择Button
self.pushButton = QtWidgets.QPushButton(self.widget2)
self.pushButton.setObjectName("pushButton")
# 横向布局容器，包含 self.label2 self.lineEdit self.pushButton
self.HBoxLayout2 = QtWidgets.QHBoxLayout(self.widget2)
self.HBoxLayout2.setContentsMargins(0, 0, 0, 0)
self.HBoxLayout2.setObjectName("HBoxLayout2")
self.HBoxLayout2.addWidget(self.label2)
self.HBoxLayout2.addWidget(self.lineEdit)
self.HBoxLayout2.addWidget(self.pushButton)
self.widget3 = QtWidgets.QWidget(imageAI)
self.widget3.setGeometry(QtCore.QRect(450, 50, 201, 401))
self.widget3.setObjectName("widget3")
# 显示识别的结果
self.label3 = QtWidgets.QLabel(self.widget3)
self.label3.setLayoutDirection(QtCore.Qt.LeftToRight)
self.label3.setWordWrap(True)
self.label3.setStyleSheet("border-width: 1px;border-style: solid;border-color: rgb(0, 0, 0);")
self.label3.setObjectName("label3")
# 复制Button
self.pushButton2 = QtWidgets.QPushButton(self.widget3)
self.pushButton2.setObjectName("pushButton2")
# 垂直布局容器，包含 self.label3 self.pushButton2
self.VBoxLayout = QtWidgets.QVBoxLayout(self.widget3)
self.VBoxLayout.setContentsMargins(0, 0, 0, 0)
self.VBoxLayout.setObjectName("VBoxLayout")
self.VBoxLayout.addWidget(self.label3)
self.VBoxLayout.addWidget(self.pushButton2)
self.retranslateUi(imageAI)
QtCore.QMetaObject.connectSlotsByName(imageAI)
def retranslateUi(self, imageAI):
_translate = QtCore.QCoreApplication.translate
imageAI.setWindowTitle(_translate("imageAI", "多媒体作业：图像识别工具 49117223柯博鑫 10417216任航"))
self.label.setText(_translate("imageAI", "选择识别类型"))
self.comboBox.setItemText(0, _translate("imageAI", "通用物体"))
self.comboBox.setItemText(1, _translate("imageAI", "菜品"))
self.comboBox.setItemText(2, _translate("imageAI", "车辆"))
self.label2.setText(_translate("imageAI", "选择要识别的图片"))
self.pushButton.setText(_translate("imageAI", "..."))
self.pushButton.clicked.connect(self.openfile)
self.label3.setText(_translate("imageAI", "显示识别结果"))
self.pushButton2.setText(_translate("imageAI", "复制结果"))
self.pushButton2.clicked.connect(self.copyText)
def openfile(self):
self.path = QFileDialog.getOpenFileName(self.widget2, "选择要识别的图片", "/", "Image Files(*.jpg *.png")
print(self.path)
if not self.path[0].strip():
pass
else:
self.lineEdit.setText(self.path[0])
pixmap = QPixmap(self.path[0])
scaledPixmap = pixmap.scaled(QtCore.QSize(311, 301), aspectRatioMode=Qt.KeepAspectRatio)
self.image.setPixmap(scaledPixmap)
self.image.show()
self.imageDetect()
pass
def copyText(self):
clipboard = QtWidgets.QApplication.clipboard()
clipboard.setText(self.label3.text())
def imageDetect(self):
'''
图片识别主函数，通过检测 comboBox 中的值来进行相应的识别
:return:
'''
# 通用物体识别
if self.comboBox.currentText() == "通用物体":
AI = AipImageClassify()
result = AI.get_general(self.path[0])
text = ''
for key, value in result[0].items():
text = text + str(key) + ':' + str(value) + '\n'
text = text + '\n'
self.label3.setText(text)
# 菜品识别
elif self.comboBox.currentText() == "菜品":
AI = AipImageClassify()
result = AI.get_dish(self.path[0])
text = ''
for key, value in result[0].items():
text = text + str(key) + ':' + str(value) + '\n'
text = text + '\n'
self.label3.setText(text)
elif self.comboBox.currentText() == "车辆":
AI = AipImageClassify()
result = AI.get_car(self.path[0])
text = ''
for key, value in result[0].items():
text = text + str(key) + ':' + str(value) + '\n'
text = text + '\n'
self.label3.setText(text)
class AipImageClassify(object):
AppID = '17817582'
API_Key = 'SGlyGbptyg7HYOMOueTlP9zd'
Secret_Key = 'qh47Wo50sGVrO9jMHxC5omUNAOU6aCDh'
client = AipImageClassify(AppID, API_Key, Secret_Key)
def get_file_content(self, filePath):
'''
读取图片
:param filePath: 图片路径
:return:
'''
with open(filePath, 'rb') as fp:
return fp.read()
def get_general(self, filePath):
'''
通用物体识别
:param filePath: 读取的图片的路径
:return: AI识别的结果
'''
image = self.get_file_content(filePath)
return self.client.advancedGeneral(image)['result']
def get_dish(self, filePath):
'''
菜品识别
:param filePath: 读取的图片的路径
:return: AI识别的结果
'''
image = self.get_file_content(filePath)
return self.client.dishDetect(image)['result']
def get_car(self, filePath):
'''
车辆识别
:param filePath: 读取的图片的路径
:return: AI识别的结果
'''
image = self.get_file_content(filePath)
return self.client.carDetect(image)['result']
if name == '__main__':
app = QtWidgets.QApplication(sys.argv)
MainWindow = QtWidgets.QMainWindow()
Ui = Ui_imageAI()
Ui.setupUi(MainWindow)
MainWindow.show()
sys.exit(app.exec_())