文章目錄

前言一、準(zhǔn)備工作1、代碼下載2、環(huán)境安裝2.1、安裝PyQt52.2、安裝Qt Designer圖形界面開發(fā)工具2.3、pycharm配置 3、模型準(zhǔn)備 二、界面展示1.界面大致如下 三、效果展示1、圖片效果展示如下：2、攝像頭或視頻檢測(cè) 四、整體代碼：總結(jié)

前言

(資料圖片)

這篇文章主要用來(lái)記錄用pyqt5搭建YOLOV5的檢測(cè)平臺(tái)，代碼是在yoloV5官方代碼上加了個(gè)qt模塊，目前可以支持GPU/CPU下pt/onnx推理。

一、準(zhǔn)備工作

1、代碼下載

2、環(huán)境安裝

根據(jù)代碼中的requirements.txt進(jìn)行環(huán)境搭建，前提是你已有Python環(huán)境

pip install -r requirements.txt

這里需要注意的是，pyqt5需要配置下環(huán)境，參考連接為：安裝pyqt5,步驟如下：

2.1、安裝PyQt5

pip install PyQt5

2.2、安裝Qt Designer圖形界面開發(fā)工具

pip install PyQt5-tools

安裝完成后，會(huì)在當(dāng)前環(huán)境目錄下生成如下文件

2.3、pycharm配置

先配置Qt Designer 參數(shù)如下： Name：Qt Designer。方便記憶。實(shí)際可以任意取值。

Program：designer.exe程序絕對(duì)路徑。根據(jù)實(shí)際安裝路徑填寫,參考路徑為安裝的xxxx/Lib/sit-packages/qt5_applications\Qt\bin\designer.exe(其中xxxx為虛擬環(huán)境目錄下)。 Working directory: $ProjectFileDir$。固定取值。

配置PyUIC，該工具是用于將Qt Designer工具開發(fā)完成的.ui文件轉(zhuǎn)化為.py文件。配置打開路徑同Qt Designer。參數(shù)配置說(shuō)明： Name：PyUIC。方便記憶。實(shí)際可以任意取值。 Program：python.exe程序絕對(duì)路徑。根據(jù)實(shí)際安裝路徑填寫。 Parameters：-m PyQt5.uic.pyuic $FileName$ -o $FileNameWithoutExtension$.py。固定取值。 Working directory: $FileDir$。固定取值。

3、模型準(zhǔn)備

根據(jù)readme.md文件中的地址選擇對(duì)應(yīng)的模型下載，如圖：

本文直接以yolov5s.pt進(jìn)行示范

二、界面展示

1.界面大致如下

代碼如下（示例）：

class Ui_Form(QtWidgets.QWidget):    def setupUi(self, Form):        Form.setObjectName("Form")        Form.resize(1130, 730)        self.stride=64        #檢測(cè)結(jié)果展示區(qū)        self.label2 = QtWidgets.QLabel(Form)        self.label2.setGeometry(QtCore.QRect(650, 410, 300, 300))        self.label2.setAlignment(QtCore.Qt.AlignCenter)        self.label2.setStyleSheet("border: 1px solid black;")        self.label2.setObjectName("label2")        #圖片或視頻路徑選擇        self.pushButton = QtWidgets.QPushButton(Form)        self.pushButton.setGeometry(QtCore.QRect(50, 310, 160, 30))        self.pushButton.setStyleSheet("font: 15pt \"宋體\";\n""border:1px solid black;\n""background-color:#e1e1e1;")        self.pushButton.setChecked(False)  # 設(shè)置開始時(shí)的狀態(tài)為未選中        self.pushButton.clicked.connect(lambda :self.select_file())        self.pushButton.setObjectName("pushButton")  #圖片選擇按鈕        #模式選擇        self.comboBox = QtWidgets.QComboBox(Form)        self.comboBox.setGeometry(QtCore.QRect(50, 260, 160, 30))        self.comboBox.setStyleSheet("font: 15pt \"宋體\";\n""border:1px solid black;\n""background-color:#e1e1e1;")        self.comboBox.setPlaceholderText("")        self.comboBox.setObjectName("comboBox")        self.comboBox.addItem("")        self.comboBox.addItem("")        self.comboBox.addItem("")        self.comboBox.addItem("")        self.comboBox.currentIndexChanged.connect(lambda :self.select_mode())        #開始檢測(cè)按鈕        self.pushButton_2 = QtWidgets.QPushButton(Form)        self.pushButton_2.setGeometry(QtCore.QRect(190, 610, 100, 70))        self.pushButton_2.setStyleSheet("font: 15pt \"宋體\";\n""border:1px solid black;\n""background-color:#e1e1e1;")        self.pushButton_2.setObjectName("pushButton_2")        self.pushButton_2.setChecked(False)        self.pushButton_2.clicked.connect(lambda :self.vim_show())        self.pushButton_3 = QtWidgets.QPushButton(Form)        self.pushButton_3.setGeometry(QtCore.QRect(50, 210, 160, 30))        self.pushButton_3.setStyleSheet("font: 15pt \"宋體\";\n""border:1px solid black;\n""background-color:#e1e1e1;")        self.pushButton_3.setObjectName("pushButton_3")        self.pushButton_3.setChecked(False) # 設(shè)置開始時(shí)的狀態(tài)為未選中        self.pushButton_3.clicked.connect(lambda :self.select_modelfile())        self.pushButton_7 = QtWidgets.QPushButton(Form)        self.pushButton_7.setGeometry(QtCore.QRect(50, 550, 160, 30))        self.pushButton_7.setStyleSheet("font: 15pt \"宋體\";\n"                                        "border:1px solid black;\n"                                        "background-color:#e1e1e1;")        self.pushButton_7.setObjectName("pushButton_7")        self.lineEdit_2=QtWidgets.QLineEdit(Form)        self.lineEdit_2.setGeometry(QtCore.QRect(230, 550, 220, 30))        self.lineEdit_2.setStyleSheet("border: 1px solid black;")        self.lineEdit_2.setObjectName("lineEdit_2")        self.label1=QtWidgets.QLabel(Form)        self.label1.setGeometry(QtCore.QRect(650, 100, 300, 300))        self.label2.setAlignment(QtCore.Qt.AlignCenter)        self.label1.setStyleSheet("border: 1px solid black;")        self.label1.setObjectName("label1")        self.textEdit = QtWidgets.QTextEdit(Form)        self.textEdit.setGeometry(QtCore.QRect(600, 100, 40, 300))        self.textEdit.setLayoutDirection(QtCore.Qt.LeftToRight)        self.textEdit.setStyleSheet("border: 1px solid black;\n""font: 15pt \"宋體\";")        self.textEdit.setObjectName("textEdit")        self.textEdit_2 = QtWidgets.QTextEdit(Form)        self.textEdit_2.setGeometry(QtCore.QRect(600, 410, 40, 300))        self.textEdit_2.setStyleSheet("border: 1px solid black;\n""font: 15pt \"宋體\";")        self.textEdit_2.setAlignment(QtCore.Qt.AlignHCenter)        self.textEdit_2.setObjectName("textEdit_2")        self.tableWidget = QtWidgets.QTableWidget(Form)        self.tableWidget.setGeometry(QtCore.QRect(230, 370, 221, 170))        self.tableWidget.setStyleSheet("font: 15pt \"宋體\";\n""")        self.tableWidget.setObjectName("tableWidget")        self.tableWidget.setColumnCount(1)        self.tableWidget.setRowCount(4)        item = QtWidgets.QTableWidgetItem()        self.tableWidget.setVerticalHeaderItem(0, item)        item = QtWidgets.QTableWidgetItem()        self.tableWidget.setVerticalHeaderItem(1, item)        item = QtWidgets.QTableWidgetItem()        self.tableWidget.setVerticalHeaderItem(2, item)        item = QtWidgets.QTableWidgetItem()        self.tableWidget.setVerticalHeaderItem(3, item)        item = QtWidgets.QTableWidgetItem()        self.tableWidget.setHorizontalHeaderItem(0, item)        self.tableWidget.horizontalHeader().setDefaultSectionSize(100)        self.tableWidget.horizontalHeader().setMinimumSectionSize(100)        self.tableWidget.verticalHeader().setDefaultSectionSize(30)        self.lineEdit_3 = QtWidgets.QLineEdit(Form)        self.lineEdit_3.setGeometry(QtCore.QRect(230, 210, 220, 30))        self.lineEdit_3.setStyleSheet("border: 1px solid black;")        self.lineEdit_3.setObjectName("lineEdit_3")        self.lineEdit_4 = QtWidgets.QLineEdit(Form)        self.lineEdit_4.setGeometry(QtCore.QRect(230, 260, 220, 30))        self.lineEdit_4.setStyleSheet("border: 1px solid black;")        self.lineEdit_4.setObjectName("lineEdit_4")        self.widget_3 = QtWidgets.QWidget(Form)        self.widget_3.setGeometry(QtCore.QRect(20, 30, 460, 690))        self.widget_3.setStyleSheet("border : 1px solid black;")        self.widget_3.setObjectName("widget_3")        self.lineEdit_5 = QtWidgets.QLineEdit(self.widget_3)        self.lineEdit_5.setGeometry(QtCore.QRect(90, 20, 261, 51))        self.lineEdit_5.setStyleSheet("font: 16pt \"宋體\";")        self.lineEdit_5.setAlignment(QtCore.Qt.AlignCenter)        self.lineEdit_5.setObjectName("lineEdit_5")        self.lineEdit_7 = QtWidgets.QLineEdit(self.widget_3)        self.lineEdit_7.setGeometry(QtCore.QRect(210, 280, 220, 30))        self.lineEdit_7.setStyleSheet("border: 1px solid black;")        self.lineEdit_7.setObjectName("lineEdit_7")        self.pushButton_4 = QtWidgets.QPushButton(self.widget_3)        self.pushButton_4.setGeometry(QtCore.QRect(30, 410, 160, 61))        self.pushButton_4.setStyleSheet("font: 15pt \"宋體\";\n""background-color:#e1e1e1;")        self.pushButton_4.setObjectName("pushButton_4")        #參數(shù)確認(rèn)按鈕        self.pushButton_5 = QtWidgets.QPushButton(Form)        self.pushButton_5.setGeometry(QtCore.QRect(50, 610, 100, 70))        self.pushButton_5.setStyleSheet("font: 15pt \"宋體\";\n""border:1px solid black;\n""background-color:#e1e1e1;")        self.pushButton_5.setObjectName("pushButton_5")        self.pushButton_5.setChecked(False)        self.pushButton_5.clicked.connect(lambda: self.param_show())        #結(jié)束檢測(cè)按鈕        self.pushButton_6 = QtWidgets.QPushButton(Form)        self.pushButton_6.setGeometry(QtCore.QRect(330, 610, 100, 70))        self.pushButton_6.setStyleSheet("font: 15pt \"宋體\";\n"                                        "border:1px solid black;\n"                                        "background-color:#e1e1e1;")        self.pushButton_6.setObjectName("pushButton_6")        self.pushButton_6.setChecked(False)        self.pushButton_6.clicked.connect(lambda: self.end_detect())        self.lineEdit_8 = QtWidgets.QLineEdit(self.widget_3)        self.lineEdit_8.setGeometry(QtCore.QRect(210, 120, 220, 30))        self.lineEdit_8.setStyleSheet("border: 1px solid black;")        self.lineEdit_8.setObjectName("lineEdit_8")        self.comboBox_2 = QtWidgets.QComboBox(self.widget_3)        self.comboBox_2.setGeometry(QtCore.QRect(30, 120, 160, 30))        self.comboBox_2.setStyleSheet("font: 15pt \"宋體\";\n""background-color:#e1e1e1;")        self.comboBox_2.setPlaceholderText("")        self.comboBox_2.setObjectName("comboBox_2")        self.comboBox_2.currentIndexChanged.connect(lambda : self.select_device())        self.comboBox_2.addItem("")        self.comboBox_2.addItem("")        self.comboBox_2.addItem("")        self.lineEdit_5.raise_()        self.lineEdit_7.raise_()        self.lineEdit_8.raise_()        self.comboBox_2.raise_()        self.pushButton_4.raise_()        self.widget_4 = QtWidgets.QWidget(Form)        self.widget_4.setGeometry(QtCore.QRect(530, 30, 510, 690))        self.widget_4.setStyleSheet("border:1px solid black;")        self.widget_4.setObjectName("widget_4")        self.lineEdit_6 = QtWidgets.QLineEdit(self.widget_4)        self.lineEdit_6.setGeometry(QtCore.QRect(70, 10, 350, 50))        self.lineEdit_6.setStyleSheet("font: 16pt \"宋體\";")        self.lineEdit_6.setAlignment(QtCore.Qt.AlignCenter)        self.lineEdit_6.setObjectName("lineEdit_6")        self.widget_4.raise_()        self.widget_3.raise_()        self.label2.raise_()        self.pushButton.raise_()        self.comboBox.raise_()        self.pushButton_3.raise_()        self.label1.raise_()        self.textEdit.raise_()        self.textEdit_2.raise_()        self.tableWidget.raise_()        self.lineEdit_3.raise_()        self.lineEdit_4.raise_()        self.pushButton_2.raise_()        self.pushButton_5.raise_()        self.pushButton_6.raise_()        self.pushButton_7.raise_()        self.lineEdit_2.raise_()        self.retranslateUi(Form)        QtCore.QMetaObject.connectSlotsByName(Form)    def retranslateUi(self, Form):        _translate = QtCore.QCoreApplication.translate        Form.setWindowTitle(_translate("Form", "YOLOV5檢測(cè)"))        self.pushButton.setText(_translate("Form", "選擇文件"))        self.pushButton_7.setText(_translate("Form", "是否結(jié)束"))        self.comboBox.setItemText(0, _translate("Form", "檢測(cè)方式"))        self.comboBox.setItemText(1, _translate("Form", "圖片檢測(cè)"))        self.comboBox.setItemText(2, _translate("Form", "視頻檢測(cè)"))        self.comboBox.setItemText(3, _translate("Form", "攝像頭檢測(cè)"))        self.pushButton_2.setText(_translate("Form", "開始檢測(cè)"))        self.pushButton_5.setText(_translate("Form", "確定"))        self.pushButton_6.setText(_translate("Form", "結(jié)束程序"))        self.pushButton_3.setText(_translate("Form", "模型路徑"))        self.textEdit.setText(_translate("Form", "\n原\n始\n圖\n片\n/\n視\n頻"))        self.textEdit_2.setText(_translate("Form", "\n\n\n測(cè)\n試\n結(jié)\n果"))        item = self.tableWidget.verticalHeaderItem(0)        item.setText(_translate("Form", "imgsize"))        item = self.tableWidget.verticalHeaderItem(1)        item.setText(_translate("Form", "conf"))        item = self.tableWidget.verticalHeaderItem(2)        item.setText(_translate("Form", "iou"))        item = self.tableWidget.verticalHeaderItem(3)        item.setText(_translate("Form", "dnn"))        item = self.tableWidget.horizontalHeaderItem(0)        item.setText(_translate("Form", "數(shù)值"))        self.tableWidget.setItem(0,0,QtWidgets.QTableWidgetItem("640")) #圖片大小默認(rèn)值        self.tableWidget.setItem(0,1,QtWidgets.QTableWidgetItem("0.25")) #置信度閾值默認(rèn)值        self.tableWidget.setItem(0,2,QtWidgets.QTableWidgetItem("0.45")) #IOU閾值默認(rèn)值        self.tableWidget.setItem(0,3,QtWidgets.QTableWidgetItem("False")) #IOU閾值默認(rèn)值        self.lineEdit_3.setPlaceholderText(_translate("Form", "模型的路徑"))        self.lineEdit_4.setPlaceholderText(_translate("Form", "圖片或視頻檢測(cè)及攝像頭檢測(cè)"))        self.lineEdit_5.setText(_translate("Form", "參數(shù)設(shè)置及可視化"))        self.lineEdit_7.setPlaceholderText(_translate("Form", "圖片或視頻路徑"))        self.pushButton_4.setText(_translate("Form", "其他參數(shù)"))        self.lineEdit_8.setPlaceholderText(_translate("Form", "運(yùn)行設(shè)備(CPU或GPU)"))        self.comboBox_2.setItemText(0, _translate("Form", "設(shè)備選擇"))        self.comboBox_2.setItemText(1, _translate("Form", "CPU"))        self.comboBox_2.setItemText(2, _translate("Form", "GPU"))        self.lineEdit_6.setText(_translate("Form", "圖片及測(cè)試結(jié)果可視化"))        self.lineEdit_2.setPlaceholderText(_translate("Form", "是否關(guān)閉視頻或者攝像頭"))

三、效果展示

1、圖片效果展示如下：

2、攝像頭或視頻檢測(cè)

四、整體代碼：

若界面太大或太小，就自己添加些代碼，將界面自適應(yīng)電腦，或者手動(dòng)修改代碼33行。

# -*- coding: utf-8 -*-# Form implementation generated from reading ui file "det.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.import sysimport osroot_path=os.path.abspath(os.path.join(os.path.dirname(__file__),".."))sys.path.append(root_path)from PyQt5 import QtCore, QtGui, QtWidgetsimport torchimport cv2from PIL import Image,ImageQtimport numpy as npfrom utils.general import non_max_suppressionfrom utils.general import check_img_size,scale_coordsfrom utils.augmentations import letterboximport osfrom pathlib import Pathfrom utils.plots import Annotator,colorsfrom models.experimental import attempt_loadclass Ui_Form(QtWidgets.QWidget):    def setupUi(self, Form):        Form.setObjectName("Form")        Form.resize(1130, 730)        self.stride=64        #檢測(cè)結(jié)果展示區(qū)        self.label2 = QtWidgets.QLabel(Form)        self.label2.setGeometry(QtCore.QRect(650, 410, 300, 300))        self.label2.setAlignment(QtCore.Qt.AlignCenter)        self.label2.setStyleSheet("border: 1px solid black;")        self.label2.setObjectName("label2")        #圖片或視頻路徑選擇        self.pushButton = QtWidgets.QPushButton(Form)        self.pushButton.setGeometry(QtCore.QRect(50, 310, 160, 30))        self.pushButton.setStyleSheet("font: 15pt \"宋體\";\n""border:1px solid black;\n""background-color:#e1e1e1;")        self.pushButton.setChecked(False)  # 設(shè)置開始時(shí)的狀態(tài)為未選中        self.pushButton.clicked.connect(lambda :self.select_file())        self.pushButton.setObjectName("pushButton")  #圖片選擇按鈕        #模式選擇        self.comboBox = QtWidgets.QComboBox(Form)        self.comboBox.setGeometry(QtCore.QRect(50, 260, 160, 30))        self.comboBox.setStyleSheet("font: 15pt \"宋體\";\n""border:1px solid black;\n""background-color:#e1e1e1;")        self.comboBox.setPlaceholderText("")        self.comboBox.setObjectName("comboBox")        self.comboBox.addItem("")        self.comboBox.addItem("")        self.comboBox.addItem("")        self.comboBox.addItem("")        self.comboBox.currentIndexChanged.connect(lambda :self.select_mode())        #開始檢測(cè)按鈕        self.pushButton_2 = QtWidgets.QPushButton(Form)        self.pushButton_2.setGeometry(QtCore.QRect(190, 610, 100, 70))        self.pushButton_2.setStyleSheet("font: 15pt \"宋體\";\n""border:1px solid black;\n""background-color:#e1e1e1;")        self.pushButton_2.setObjectName("pushButton_2")        self.pushButton_2.setChecked(False)        self.pushButton_2.clicked.connect(lambda :self.vim_show())        self.pushButton_3 = QtWidgets.QPushButton(Form)        self.pushButton_3.setGeometry(QtCore.QRect(50, 210, 160, 30))        self.pushButton_3.setStyleSheet("font: 15pt \"宋體\";\n""border:1px solid black;\n""background-color:#e1e1e1;")        self.pushButton_3.setObjectName("pushButton_3")        self.pushButton_3.setChecked(False) # 設(shè)置開始時(shí)的狀態(tài)為未選中        self.pushButton_3.clicked.connect(lambda :self.select_modelfile())        self.pushButton_7 = QtWidgets.QPushButton(Form)        self.pushButton_7.setGeometry(QtCore.QRect(50, 550, 160, 30))        self.pushButton_7.setStyleSheet("font: 15pt \"宋體\";\n"                                        "border:1px solid black;\n"                                        "background-color:#e1e1e1;")        self.pushButton_7.setObjectName("pushButton_7")        self.lineEdit_2=QtWidgets.QLineEdit(Form)        self.lineEdit_2.setGeometry(QtCore.QRect(230, 550, 220, 30))        self.lineEdit_2.setStyleSheet("border: 1px solid black;")        self.lineEdit_2.setObjectName("lineEdit_2")        self.label1=QtWidgets.QLabel(Form)        self.label1.setGeometry(QtCore.QRect(650, 100, 300, 300))        self.label2.setAlignment(QtCore.Qt.AlignCenter)        self.label1.setStyleSheet("border: 1px solid black;")        self.label1.setObjectName("label1")        self.textEdit = QtWidgets.QTextEdit(Form)        self.textEdit.setGeometry(QtCore.QRect(600, 100, 40, 300))        self.textEdit.setLayoutDirection(QtCore.Qt.LeftToRight)        self.textEdit.setStyleSheet("border: 1px solid black;\n""font: 15pt \"宋體\";")        self.textEdit.setObjectName("textEdit")        self.textEdit_2 = QtWidgets.QTextEdit(Form)        self.textEdit_2.setGeometry(QtCore.QRect(600, 410, 40, 300))        self.textEdit_2.setStyleSheet("border: 1px solid black;\n""font: 15pt \"宋體\";")        self.textEdit_2.setAlignment(QtCore.Qt.AlignHCenter)        self.textEdit_2.setObjectName("textEdit_2")        self.tableWidget = QtWidgets.QTableWidget(Form)        self.tableWidget.setGeometry(QtCore.QRect(230, 370, 221, 170))        self.tableWidget.setStyleSheet("font: 15pt \"宋體\";\n""")        self.tableWidget.setObjectName("tableWidget")        self.tableWidget.setColumnCount(1)        self.tableWidget.setRowCount(4)        item = QtWidgets.QTableWidgetItem()        self.tableWidget.setVerticalHeaderItem(0, item)        item = QtWidgets.QTableWidgetItem()        self.tableWidget.setVerticalHeaderItem(1, item)        item = QtWidgets.QTableWidgetItem()        self.tableWidget.setVerticalHeaderItem(2, item)        item = QtWidgets.QTableWidgetItem()        self.tableWidget.setVerticalHeaderItem(3, item)        item = QtWidgets.QTableWidgetItem()        self.tableWidget.setHorizontalHeaderItem(0, item)        self.tableWidget.horizontalHeader().setDefaultSectionSize(100)        self.tableWidget.horizontalHeader().setMinimumSectionSize(100)        self.tableWidget.verticalHeader().setDefaultSectionSize(30)        self.lineEdit_3 = QtWidgets.QLineEdit(Form)        self.lineEdit_3.setGeometry(QtCore.QRect(230, 210, 220, 30))        self.lineEdit_3.setStyleSheet("border: 1px solid black;")        self.lineEdit_3.setObjectName("lineEdit_3")        self.lineEdit_4 = QtWidgets.QLineEdit(Form)        self.lineEdit_4.setGeometry(QtCore.QRect(230, 260, 220, 30))        self.lineEdit_4.setStyleSheet("border: 1px solid black;")        self.lineEdit_4.setObjectName("lineEdit_4")        self.widget_3 = QtWidgets.QWidget(Form)        self.widget_3.setGeometry(QtCore.QRect(20, 30, 460, 690))        self.widget_3.setStyleSheet("border : 1px solid black;")        self.widget_3.setObjectName("widget_3")        self.lineEdit_5 = QtWidgets.QLineEdit(self.widget_3)        self.lineEdit_5.setGeometry(QtCore.QRect(90, 20, 261, 51))        self.lineEdit_5.setStyleSheet("font: 16pt \"宋體\";")        self.lineEdit_5.setAlignment(QtCore.Qt.AlignCenter)        self.lineEdit_5.setObjectName("lineEdit_5")        self.lineEdit_7 = QtWidgets.QLineEdit(self.widget_3)        self.lineEdit_7.setGeometry(QtCore.QRect(210, 280, 220, 30))        self.lineEdit_7.setStyleSheet("border: 1px solid black;")        self.lineEdit_7.setObjectName("lineEdit_7")        self.pushButton_4 = QtWidgets.QPushButton(self.widget_3)        self.pushButton_4.setGeometry(QtCore.QRect(30, 410, 160, 61))        self.pushButton_4.setStyleSheet("font: 15pt \"宋體\";\n""background-color:#e1e1e1;")        self.pushButton_4.setObjectName("pushButton_4")        #參數(shù)確認(rèn)按鈕        self.pushButton_5 = QtWidgets.QPushButton(Form)        self.pushButton_5.setGeometry(QtCore.QRect(50, 610, 100, 70))        self.pushButton_5.setStyleSheet("font: 15pt \"宋體\";\n""border:1px solid black;\n""background-color:#e1e1e1;")        self.pushButton_5.setObjectName("pushButton_5")        self.pushButton_5.setChecked(False)        self.pushButton_5.clicked.connect(lambda: self.param_show())        #結(jié)束檢測(cè)按鈕        self.pushButton_6 = QtWidgets.QPushButton(Form)        self.pushButton_6.setGeometry(QtCore.QRect(330, 610, 100, 70))        self.pushButton_6.setStyleSheet("font: 15pt \"宋體\";\n"                                        "border:1px solid black;\n"                                        "background-color:#e1e1e1;")        self.pushButton_6.setObjectName("pushButton_6")        self.pushButton_6.setChecked(False)        self.pushButton_6.clicked.connect(lambda: self.end_detect())        self.lineEdit_8 = QtWidgets.QLineEdit(self.widget_3)        self.lineEdit_8.setGeometry(QtCore.QRect(210, 120, 220, 30))        self.lineEdit_8.setStyleSheet("border: 1px solid black;")        self.lineEdit_8.setObjectName("lineEdit_8")        self.comboBox_2 = QtWidgets.QComboBox(self.widget_3)        self.comboBox_2.setGeometry(QtCore.QRect(30, 120, 160, 30))        self.comboBox_2.setStyleSheet("font: 15pt \"宋體\";\n""background-color:#e1e1e1;")        self.comboBox_2.setPlaceholderText("")        self.comboBox_2.setObjectName("comboBox_2")        self.comboBox_2.currentIndexChanged.connect(lambda : self.select_device())        self.comboBox_2.addItem("")        self.comboBox_2.addItem("")        self.comboBox_2.addItem("")        self.lineEdit_5.raise_()        self.lineEdit_7.raise_()        self.lineEdit_8.raise_()        self.comboBox_2.raise_()        self.pushButton_4.raise_()        self.widget_4 = QtWidgets.QWidget(Form)        self.widget_4.setGeometry(QtCore.QRect(530, 30, 510, 690))        self.widget_4.setStyleSheet("border:1px solid black;")        self.widget_4.setObjectName("widget_4")        self.lineEdit_6 = QtWidgets.QLineEdit(self.widget_4)        self.lineEdit_6.setGeometry(QtCore.QRect(70, 10, 350, 50))        self.lineEdit_6.setStyleSheet("font: 16pt \"宋體\";")        self.lineEdit_6.setAlignment(QtCore.Qt.AlignCenter)        self.lineEdit_6.setObjectName("lineEdit_6")        self.widget_4.raise_()        self.widget_3.raise_()        self.label2.raise_()        self.pushButton.raise_()        self.comboBox.raise_()        self.pushButton_3.raise_()        self.label1.raise_()        self.textEdit.raise_()        self.textEdit_2.raise_()        self.tableWidget.raise_()        self.lineEdit_3.raise_()        self.lineEdit_4.raise_()        self.pushButton_2.raise_()        self.pushButton_5.raise_()        self.pushButton_6.raise_()        self.pushButton_7.raise_()        self.lineEdit_2.raise_()        self.retranslateUi(Form)        QtCore.QMetaObject.connectSlotsByName(Form)    def retranslateUi(self, Form):        _translate = QtCore.QCoreApplication.translate        Form.setWindowTitle(_translate("Form", "YOLOV5檢測(cè)"))        self.pushButton.setText(_translate("Form", "選擇文件"))        self.pushButton_7.setText(_translate("Form", "是否結(jié)束"))        self.comboBox.setItemText(0, _translate("Form", "檢測(cè)方式"))        self.comboBox.setItemText(1, _translate("Form", "圖片檢測(cè)"))        self.comboBox.setItemText(2, _translate("Form", "視頻檢測(cè)"))        self.comboBox.setItemText(3, _translate("Form", "攝像頭檢測(cè)"))        self.pushButton_2.setText(_translate("Form", "開始檢測(cè)"))        self.pushButton_5.setText(_translate("Form", "確定"))        self.pushButton_6.setText(_translate("Form", "結(jié)束程序"))        self.pushButton_3.setText(_translate("Form", "模型路徑"))        self.textEdit.setText(_translate("Form", "\n原\n始\n圖\n片\n/\n視\n頻"))        self.textEdit_2.setText(_translate("Form", "\n\n\n測(cè)\n試\n結(jié)\n果"))        item = self.tableWidget.verticalHeaderItem(0)        item.setText(_translate("Form", "imgsize"))        item = self.tableWidget.verticalHeaderItem(1)        item.setText(_translate("Form", "conf"))        item = self.tableWidget.verticalHeaderItem(2)        item.setText(_translate("Form", "iou"))        item = self.tableWidget.verticalHeaderItem(3)        item.setText(_translate("Form", "dnn"))        item = self.tableWidget.horizontalHeaderItem(0)        item.setText(_translate("Form", "數(shù)值"))        self.tableWidget.setItem(0,0,QtWidgets.QTableWidgetItem("640")) #圖片大小默認(rèn)值        self.tableWidget.setItem(0,1,QtWidgets.QTableWidgetItem("0.25")) #置信度閾值默認(rèn)值        self.tableWidget.setItem(0,2,QtWidgets.QTableWidgetItem("0.45")) #IOU閾值默認(rèn)值        self.tableWidget.setItem(0,3,QtWidgets.QTableWidgetItem("False")) #IOU閾值默認(rèn)值        self.lineEdit_3.setPlaceholderText(_translate("Form", "模型的路徑"))        self.lineEdit_4.setPlaceholderText(_translate("Form", "圖片或視頻檢測(cè)及攝像頭檢測(cè)"))        self.lineEdit_5.setText(_translate("Form", "參數(shù)設(shè)置及可視化"))        self.lineEdit_7.setPlaceholderText(_translate("Form", "圖片或視頻路徑"))        self.pushButton_4.setText(_translate("Form", "其他參數(shù)"))        self.lineEdit_8.setPlaceholderText(_translate("Form", "運(yùn)行設(shè)備(CPU或GPU)"))        self.comboBox_2.setItemText(0, _translate("Form", "設(shè)備選擇"))        self.comboBox_2.setItemText(1, _translate("Form", "CPU"))        self.comboBox_2.setItemText(2, _translate("Form", "GPU"))        self.lineEdit_6.setText(_translate("Form", "圖片及測(cè)試結(jié)果可視化"))        self.lineEdit_2.setPlaceholderText(_translate("Form", "是否關(guān)閉視頻或者攝像頭"))    #選擇圖片或視頻路徑    def select_file(self):        file,_=QtWidgets.QFileDialog.getOpenFileName(None,"選擇文件",os.getcwd(),"All Files(*.jpg;*.png;*.jpeg;*.mp4;*.flv;*.avi;);;Text Files (*.txt)")        self.lineEdit_7.setText(file)    #選擇模型路徑    def select_modelfile(self):        file,_=QtWidgets.QFileDialog.getOpenFileName(None,"選擇文件",os.getcwd(),"All Files(*.pt;*.onnx;*.pth);;")        self.lineEdit_3.setText(file)    #設(shè)備選擇    def select_device(self):        device=self.comboBox_2.currentText() #獲取下拉框的信息        self.lineEdit_8.setText(device)    #模式選擇    def select_mode(self):        mode=self.comboBox.currentText()        self.lineEdit_4.setText(mode)    #模型初始化    def param_show(self):        device = torch.device("cpu" if self.lineEdit_8.text()=="CPU" else "cuda") #設(shè)備        modelpath = self.lineEdit_3.text() #模型路徑        self.suffix = Path(modelpath).suffix[1:]        if self.suffix == "pt":            self.flag=True            self.model = attempt_load(modelpath if isinstance(modelpath, list) else modelpath,                                 map_location=device)            self.stride = int(self.model.stride.max())  # model stride            self.names = self.model.module.names if hasattr(self.model, "module") else self.model.names  # get class names            if self.lineEdit_8.text() == "GPU":                self.model.half()        elif self.suffix == "onnx":            self.dnn = False            self.flag=False            if self.dnn:                # check_requirements(("opencv-python>=4.5.4",))                self.model = cv2.dnn.readNetFromONNX(modelpath)            else:                # check_requirements(("onnx", "onnxruntime-gpu" if torch.has_cuda else "onnxruntime"))                import onnxruntime                self.model = onnxruntime.InferenceSession(modelpath, None)        # 后兩項(xiàng)分別為按鈕(以|隔開，共有7種按鈕類型，見(jiàn)示例后)、默認(rèn)按鈕(省略則默認(rèn)為第一個(gè)按鈕)        msg=QtWidgets.QMessageBox(QtWidgets.QMessageBox.Warning,"消息框", "模型初始化完成,請(qǐng)開始檢測(cè)")        msg.exec_()    #檢測(cè)    def vim_show(self):        filepath=self.lineEdit_7.text() #文件路徑        detect_mode=self.lineEdit_4.text()#檢測(cè)模式        imgsize = int(self.tableWidget.item(0, 0).text())  # 圖片尺寸        conf_thres = float(self.tableWidget.item(1, 0).text())  # 置信度閾值        iou_thres = float(self.tableWidget.item(2, 0).text())  # iou閾值        device = self.lineEdit_8.text()  # 設(shè)備        if detect_mode=="圖片檢測(cè)":            #opencv讀圖展示            origin_img=cv2.imread(filepath)            if origin_img.shape[0] > origin_img.shape[1]:  # h>w                orgin_img1 = cv2.resize(origin_img, (int(origin_img.shape[1] * (300 / origin_img.shape[0])-1), 300),                                 cv2.INTER_LINEAR)            else:  # w>=h                orgin_img1 = cv2.resize(origin_img, (300, int(origin_img.shape[0] * (300 / origin_img.shape[1]))-1),                                 cv2.INTER_LINEAR)            #圖片預(yù)處理            input_img=self.process_img(origin_img,device,imgsize)            #推理            result=self.process_inference(origin_img,input_img,conf_thres,iou_thres)            #結(jié)果展示            if result.shape[0] > result.shape[1]:  # h>w                img1 = cv2.resize(result, (int(result.shape[1] * (300 / result.shape[0]))-1, 300),                                 cv2.INTER_LINEAR)            else:  # w>=h                img1 = cv2.resize(result, (300, int(result.shape[0] * (300 / result.shape[1]))-1),                                 cv2.INTER_LINEAR)            #原圖展示            rgb_img = cv2.cvtColor(orgin_img1, cv2.COLOR_BGR2RGB)            qimg = QtGui.QImage(rgb_img.data, rgb_img.shape[1], rgb_img.shape[0], QtGui.QImage.Format_RGB888)            self.label1.setPixmap(QtGui.QPixmap.fromImage(qimg))            #結(jié)果展示            rgb_img1 = cv2.cvtColor(img1, cv2.COLOR_BGR2RGB)            qimg1 = QtGui.QImage(rgb_img1.data, rgb_img1.shape[1], rgb_img1.shape[0], QtGui.QImage.Format_RGB888)            self.label2.setPixmap(QtGui.QPixmap.fromImage(qimg1))        elif detect_mode=="視頻檢測(cè)":            cap = cv2.VideoCapture(filepath)            count = 1            while cap.isOpened():                if count % 2 == 0:                    ret, frame = cap.read()                    if ret:                        if frame.shape[0] > frame.shape[1]:  # h>w                            img = cv2.resize(frame, (int(frame.shape[1] * (300 / frame.shape[0])), 300),                                             cv2.INTER_LINEAR)                        else:  # w>=h                            img = cv2.resize(frame, (300, int(frame.shape[0] * (300 / frame.shape[1]))),                                             cv2.INTER_LINEAR)                        input_img = self.process_img(frame, device, imgsize)                        result = self.process_inference(frame, input_img, conf_thres, iou_thres)                        if result.shape[0] > result.shape[1]:  # h>w                            img1 = cv2.resize(result, (int(result.shape[1] * (300 / result.shape[0])) - 1, 300),                                              cv2.INTER_LINEAR)                        else:  # h<=w 1="" 2="=" 300="" img1="cv2.resize(result," -="" img="cv2.cvtColor(img," qimg="QtGui.QImage(img.data," rgb1="cv2.cvtColor(img1," qimg1="QtGui.QImage(rgb1.data," else:="" break="" count="" self.cap="cv2.VideoCapture(0)" while="" true:="" if="" 0:="" frame="self.cap.read()" ret:=""> frame.shape[1]:  # h>w                            img = cv2.resize(frame, (int(frame.shape[1] * (300 / frame.shape[0])), 300),                                             cv2.INTER_LINEAR)                        else:  # w>=h                            img = cv2.resize(frame, (300, int(frame.shape[0] * (300 / frame.shape[1]))),                                             cv2.INTER_LINEAR)                        input_img=self.process_img(frame,device,imgsize)                        result=self.process_inference(frame,input_img,conf_thres,iou_thres)                        if result.shape[0]>result.shape[1]: #h>w                            img1=cv2.resize(result,(int(result.shape[1]*(300/result.shape[0]))-1,300),cv2.INTER_LINEAR)                        else: #h<=w                            img1=cv2.resize(result,(300,int(result.shape[0]*(300/result.shape[1]))-1),                                            cv2.INTER_LINEAR)                        img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)                        cv2.waitKey(1)                        qimg = QtGui.QImage(img.data, img.shape[1], img.shape[0], QtGui.QImage.Format_RGB888)                        self.label1.setPixmap(QtGui.QPixmap.fromImage(qimg))                        #結(jié)果展示                        rgb1=cv2.cvtColor(img1,cv2.COLOR_BGR2RGB)                        qimg1=QtGui.QImage(rgb1.data,rgb1.shape[1],rgb1.shape[0],QtGui.QImage.Format_RGB888)                        self.label2.setPixmap(QtGui.QPixmap.fromImage(qimg1))                    #復(fù)位                    count=1                    if self.lineEdit_2.text()=="是":                        break                else:                    count += 1            self.cap.release()            cv2.destroyAllWindows()    #退出程序    def end_detect(self):       sys.exit(1)    #圖片處理    def process_img(self,org_img,device,imgsize):        """        Args:            org_img: 原圖            device: 設(shè)備            imgsize: 圖片尺寸        Returns:        """        # 查看圖片尺寸是否能別stride整除        imgsz = check_img_size(imgsize, s=self.stride)        # resize_padding        img = letterbox(org_img, imgsz, self.stride, auto=self.flag)[0]        img = img.transpose((2, 0, 1))[::-1]  # HWC to CHW, BGR to RGB        img = np.ascontiguousarray(img)        if self.suffix == "onnx":            img = img.astype("float32")        else:            img = torch.from_numpy(img).to(device.lower())            # 設(shè)置是否為半精度測(cè)試            if device == "GPU":                img = img.half()            else:                img=img.float()        img /= 255.0  # 歸一化        if len(img.shape) == 3:            img = img[None]        return img    #推理進(jìn)程    def process_inference(self,org_img,img,conf_thres,iou_thres):        """        Args:            org_img: 原圖            img: 縮放后的圖            conf_thres: 置信度閾值            iou_thres: IOU閾值        Returns:        """        if self.suffix == "pt":            pred = self.model(img)[0]        elif self.suffix == "onnx":            if self.dnn:                self.model.setInput(img)                pred = torch.tensor(self.model.forward())            else:                pred = torch.tensor(self.model.run([self.model.get_outputs()[0].name], {self.model.get_inputs()[0].name: img}))        pred = non_max_suppression(pred, conf_thres, iou_thres, classes=None, agnostic=False, max_det=1000)        dt, seen = [0.0, 0.0, 0.0], 0        for i, det in enumerate(pred):            seen += 1            im0 = org_img.copy()            # 繪制預(yù)測(cè)框            annotator = Annotator(im0, line_width=3, example=str(1))            # 檢測(cè)到目標(biāo)時(shí)            if len(det):                det[:, :4] = scale_coords(img.shape[2:], det[:, :4], im0.shape).round()                # 寫結(jié)果                for *xyxy, conf, cls in reversed(det):                    c = int(cls)                    # label = f"{conf:.2f}"                    label = f"{self.names[c]} {conf:.2f}"                    annotator.box_label(xyxy, label, color=colors(c, True))            im0 = annotator.result()            # cv2.imwrite(save_path + os.sep + Path(img_path).name, im0)        return im0if __name__ == "__main__":    import sys    QtCore.QCoreApplication.setAttribute(QtCore.Qt.AA_EnableHighDpiScaling)  # 解決了Qtdesigner設(shè)計(jì)的界面與實(shí)際運(yùn)行界面不一致的問(wèn)題    app = QtWidgets.QApplication(sys.argv)    widget = QtWidgets.QWidget()    ui = Ui_Form()    ui.setupUi(widget)    widget.show()    sys.exit(app.exec_())

總結(jié)

以上就是本篇文章的全部?jī)?nèi)容，歡迎評(píng)論區(qū)指正錯(cuò)誤，也可加入QQ群：995760755交流。