• Yolov5(v5.0) + pyqt5界面设计

    1.下载安装pyqt5工具包以及配置ui界面开发环境

    1. pip install PyQt5
    2. pip install PyQt5-tools

    2.点击File->Settings->External Tools进行工具添加，依次进行Qt Designer、PyUIC环境配置.

     2.1 添加QtDesigner

     Qt Designer 是通过拖拽的方式放置控件，并实时查看控件效果进行快速UI设计

    位置内容
    name可以随便命名，只要便于记忆就可以，本次采取通用命名：Qt Designer
    Programdesigner.exe路径,一般在python中.\Library\bin\designer.exe
    Arguments固定格式，直接复制也可：$FileDir$\$FileName$
    Working directory固定格式，直接复制也可：$FileDir$

    2.2 添加PyUIC

     PyUIC主要是把Qt Designer生成的.ui文件换成.py文件

    位置内容
    name可以随便命名，只要便于记忆就可以，本次采取通用命名：PyUiC
    Programpython.exe路径,一般在python安装根目录中
    Arguments固定格式，直接复制也可：-m PyQt5.uic.pyuic $FileName$ -o $FileNameWithoutExtension$.py
    Working directory固定格式，直接复制也可：$FileDir$

    3. QtDesigner建立图形化窗口界面 

    3.1 在根目录下新建UI文件夹进行UI文件的专门存储，点击Tools->External Tools->Qt Designer进行图形界面创建.

     3.2 创建一个Main Window窗口

    3.3 完成基本界面开发后，保存其为Detect.ui，放置在UI文件夹下，利用PyUic工具将其转化为Detect.py文件。

    转换完成后，进行相应的槽函数的建立与修改，此处建议直接看我后面给出的demo。

    4. demo

    使用时只需将parser.add_argument中的'--weights'设为响应权重即可。

    1. # -*- coding: utf-8 -*-
    2.  
    3. # Form implementation generated from reading ui file '.\project.ui'
    4. #
    5. # Created by: PyQt5 UI code generator 5.9.2
    6. #
    7. # WARNING! All changes made in this file will be lost!
    8. import sys
    9. import cv2
    10. import argparse
    11. import random
    12. import torch
    13. import numpy as np
    14. import torch.backends.cudnn as cudnn
    15.  
    16. from PyQt5 import QtCore, QtGui, QtWidgets
    17.  
    18. from utils.torch_utils import select_device
    19. from models.experimental import attempt_load
    20. from utils.general import check_img_size, non_max_suppression, scale_coords
    21. from utils.datasets import letterbox
    22. from utils.plots import plot_one_box
    23.  
    24.  
    25. class Ui_MainWindow(QtWidgets.QMainWindow):
    26.     def __init__(self, parent=None):
    27.         super(Ui_MainWindow, self).__init__(parent)
    28.         self.timer_video = QtCore.QTimer()
    29.         self.setupUi(self)
    30.         self.init_logo()
    31.         self.init_slots()
    32.         self.cap = cv2.VideoCapture()
    33.         self.out = None
    34.         # self.out = cv2.VideoWriter('prediction.avi', cv2.VideoWriter_fourcc(*'XVID'), 20.0, (640, 480))
    35.  
    36.         parser = argparse.ArgumentParser()
    37.         parser.add_argument('--weights', nargs='+', type=str,
    38.                             default='weights/best.pt', help='model.pt path(s)')
    39.         # file/folder, 0 for webcam
    40.         parser.add_argument('--source', type=str,
    41.                             default='data/images', help='source')
    42.         parser.add_argument('--img-size', type=int,
    43.                             default=640, help='inference size (pixels)')
    44.         parser.add_argument('--conf-thres', type=float,
    45.                             default=0.25, help='object confidence threshold')
    46.         parser.add_argument('--iou-thres', type=float,
    47.                             default=0.45, help='IOU threshold for NMS')
    48.         parser.add_argument('--device', default='',
    49.                             help='cuda device, i.e. 0 or 0,1,2,3 or cpu')
    50.         parser.add_argument(
    51.             '--view-img', action='store_true', help='display results')
    52.         parser.add_argument('--save-txt', action='store_true',
    53.                             help='save results to *.txt')
    54.         parser.add_argument('--save-conf', action='store_true',
    55.                             help='save confidences in --save-txt labels')
    56.         parser.add_argument('--nosave', action='store_true',
    57.                             help='do not save images/videos')
    58.         parser.add_argument('--classes', nargs='+', type=int,
    59.                             help='filter by class: --class 0, or --class 0 2 3')
    60.         parser.add_argument(
    61.             '--agnostic-nms', action='store_true', help='class-agnostic NMS')
    62.         parser.add_argument('--augment', action='store_true',
    63.                             help='augmented inference')
    64.         parser.add_argument('--update', action='store_true',
    65.                             help='update all models')
    66.         parser.add_argument('--project', default='runs/detect',
    67.                             help='save results to project/name')
    68.         parser.add_argument('--name', default='exp',
    69.                             help='save results to project/name')
    70.         parser.add_argument('--exist-ok', action='store_true',
    71.                             help='existing project/name ok, do not increment')
    72.         self.opt = parser.parse_args()
    73.         print(self.opt)
    74.  
    75.         source, weights, view_img, save_txt, imgsz = self.opt.source, self.opt.weights, self.opt.view_img, self.opt.save_txt, self.opt.img_size
    76.  
    77.         self.device = select_device(self.opt.device)
    78.         self.half = self.device.type != 'cpu'  # half precision only supported on CUDA
    79.  
    80.         cudnn.benchmark = True
    81.  
    82.         # Load model
    83.         self.model = attempt_load(
    84.             weights, map_location=self.device)  # load FP32 model
    85.         stride = int(self.model.stride.max())  # model stride
    86.         self.imgsz = check_img_size(imgsz, s=stride)  # check img_size
    87.         if self.half:
    88.             self.model.half()  # to FP16
    89.  
    90.         # Get names and colors
    91.         self.names = self.model.module.names if hasattr(
    92.             self.model, 'module') else self.model.names
    93.         self.colors = [[random.randint(0, 255)
    94.                         for _ in range(3)] for _ in self.names]
    95.  
    96.     def setupUi(self, MainWindow):
    97.         MainWindow.setObjectName("MainWindow")
    98.         MainWindow.resize(800, 600)
    99.         self.centralwidget = QtWidgets.QWidget(MainWindow)
    100.         self.centralwidget.setObjectName("centralwidget")
    101.         self.pushButton = QtWidgets.QPushButton(self.centralwidget)
    102.         self.pushButton.setGeometry(QtCore.QRect(20, 130, 112, 34))
    103.         self.pushButton.setObjectName("pushButton")
    104.         self.pushButton_2 = QtWidgets.QPushButton(self.centralwidget)
    105.         self.pushButton_2.setGeometry(QtCore.QRect(20, 220, 112, 34))
    106.         self.pushButton_2.setObjectName("pushButton_2")
    107.         self.pushButton_3 = QtWidgets.QPushButton(self.centralwidget)
    108.         self.pushButton_3.setGeometry(QtCore.QRect(20, 300, 112, 34))
    109.         self.pushButton_3.setObjectName("pushButton_3")
    110.         self.groupBox = QtWidgets.QGroupBox(self.centralwidget)
    111.         self.groupBox.setGeometry(QtCore.QRect(160, 90, 611, 411))
    112.         self.groupBox.setObjectName("groupBox")
    113.         self.label = QtWidgets.QLabel(self.groupBox)
    114.         self.label.setGeometry(QtCore.QRect(10, 40, 561, 331))
    115.         self.label.setObjectName("label")
    116.         self.textEdit = QtWidgets.QTextEdit(self.centralwidget)
    117.         self.textEdit.setGeometry(QtCore.QRect(150, 10, 471, 51))
    118.         self.textEdit.setObjectName("textEdit")
    119.         MainWindow.setCentralWidget(self.centralwidget)
    120.         self.menubar = QtWidgets.QMenuBar(MainWindow)
    121.         self.menubar.setGeometry(QtCore.QRect(0, 0, 800, 30))
    122.         self.menubar.setObjectName("menubar")
    123.         MainWindow.setMenuBar(self.menubar)
    124.         self.statusbar = QtWidgets.QStatusBar(MainWindow)
    125.         self.statusbar.setObjectName("statusbar")
    126.         MainWindow.setStatusBar(self.statusbar)
    127.  
    128.         self.retranslateUi(MainWindow)
    129.         QtCore.QMetaObject.connectSlotsByName(MainWindow)
    130.  
    131.     def retranslateUi(self, MainWindow):
    132.         _translate = QtCore.QCoreApplication.translate
    133.         MainWindow.setWindowTitle(_translate("MainWindow", "演示系统"))
    134.         self.pushButton.setText(_translate("MainWindow", "图片检测"))
    135.         self.pushButton_2.setText(_translate("MainWindow", "摄像头检测"))
    136.         self.pushButton_3.setText(_translate("MainWindow", "视频检测"))
    137.         self.groupBox.setTitle(_translate("MainWindow", "检测结果"))
    138.         self.label.setText(_translate("MainWindow", "TextLabel"))
    139.         self.textEdit.setHtml(_translate("MainWindow",
    140.             "\n"
    141.             "\n"
    142.             "
      演示系统
      "      ))
    143.  
    144.     def init_slots(self):
    145.         self.pushButton.clicked.connect(self.button_image_open)
    146.         self.pushButton_3.clicked.connect(self.button_video_open)
    147.         self.pushButton_2.clicked.connect(self.button_camera_open)
    148.         self.timer_video.timeout.connect(self.show_video_frame)
    149.  
    150.     def init_logo(self):
    151.         pix = QtGui.QPixmap('wechat.jpg')
    152.         self.label.setScaledContents(True)
    153.         self.label.setPixmap(pix)
    154.  
    155.     def button_image_open(self):
    156.         print('button_image_open')
    157.         name_list = []
    158.  
    159.         img_name, _ = QtWidgets.QFileDialog.getOpenFileName(
    160.             self, "打开图片", "", "*.jpg;;*.png;;All Files(*)")
    161.         if not img_name:
    162.             return
    163.  
    164.         img = cv2.imread(img_name)
    165.         print(img_name)
    166.         showimg = img
    167.         with torch.no_grad():
    168.             img = letterbox(img, new_shape=self.opt.img_size)[0]
    169.             # Convert
    170.             # BGR to RGB, to 3x416x416
    171.             img = img[:, :, ::-1].transpose(2, 0, 1)
    172.             img = np.ascontiguousarray(img)
    173.             img = torch.from_numpy(img).to(self.device)
    174.             img = img.half() if self.half else img.float()  # uint8 to fp16/32
    175.             img /= 255.0  # 0 - 255 to 0.0 - 1.0
    176.             if img.ndimension() == 3:
    177.                 img = img.unsqueeze(0)
    178.             # Inference
    179.             pred = self.model(img, augment=self.opt.augment)[0]
    180.             # Apply NMS
    181.             pred = non_max_suppression(pred, self.opt.conf_thres, self.opt.iou_thres, classes=self.opt.classes,
    182.                                        agnostic=self.opt.agnostic_nms)
    183.             print(pred)
    184.             # Process detections
    185.             for i, det in enumerate(pred):
    186.                 if det is not None and len(det):
    187.                     # Rescale boxes from img_size to im0 size
    188.                     det[:, :4] = scale_coords(
    189.                         img.shape[2:], det[:, :4], showimg.shape).round()
    190.  
    191.                     for *xyxy, conf, cls in reversed(det):
    192.                         label = '%s %.2f' % (self.names[int(cls)], conf)
    193.                         name_list.append(self.names[int(cls)])
    194.                         plot_one_box(xyxy, showimg, label=label,
    195.                                      color=self.colors[int(cls)], line_thickness=2)
    196.  
    197.         cv2.imwrite('prediction.jpg', showimg)
    198.         self.result = cv2.cvtColor(showimg, cv2.COLOR_BGR2BGRA)
    199.         self.result = cv2.resize(
    200.             self.result, (640, 480), interpolation=cv2.INTER_AREA)
    201.         self.QtImg = QtGui.QImage(
    202.             self.result.data, self.result.shape[1], self.result.shape[0], QtGui.QImage.Format_RGB32)
    203.         self.label.setPixmap(QtGui.QPixmap.fromImage(self.QtImg))
    204.  
    205.     def button_video_open(self):
    206.         video_name, _ = QtWidgets.QFileDialog.getOpenFileName(
    207.             self, "打开视频", "", "*.mp4;;*.avi;;All Files(*)")
    208.  
    209.         if not video_name:
    210.             return
    211.  
    212.         flag = self.cap.open(video_name)
    213.         if flag == False:
    214.             QtWidgets.QMessageBox.warning(
    215.                 self, u"Warning", u"打开视频失败", buttons=QtWidgets.QMessageBox.Ok, defaultButton=QtWidgets.QMessageBox.Ok)
    216.         else:
    217.             self.out = cv2.VideoWriter('prediction.avi', cv2.VideoWriter_fourcc(
    218.                 *'MJPG'), 20, (int(self.cap.get(3)), int(self.cap.get(4))))
    219.             self.timer_video.start(30)
    220.             self.pushButton_3.setDisabled(True)
    221.             self.pushButton.setDisabled(True)
    222.             self.pushButton_2.setDisabled(True)
    223.  
    224.     def button_camera_open(self):
    225.         if not self.timer_video.isActive():
    226.             # 默认使用第一个本地camera
    227.             flag = self.cap.open(0)
    228.             if flag == False:
    229.                 QtWidgets.QMessageBox.warning(
    230.                     self, u"Warning", u"打开摄像头失败", buttons=QtWidgets.QMessageBox.Ok,
    231.                     defaultButton=QtWidgets.QMessageBox.Ok)
    232.             else:
    233.                 self.out = cv2.VideoWriter('prediction.avi', cv2.VideoWriter_fourcc(
    234.                     *'MJPG'), 20, (int(self.cap.get(3)), int(self.cap.get(4))))
    235.                 self.timer_video.start(30)
    236.                 self.pushButton_3.setDisabled(True)
    237.                 self.pushButton.setDisabled(True)
    238.                 self.pushButton_2.setText(u"关闭摄像头")
    239.         else:
    240.             self.timer_video.stop()
    241.             self.cap.release()
    242.             self.out.release()
    243.             self.label.clear()
    244.             self.init_logo()
    245.             self.pushButton_3.setDisabled(False)
    246.             self.pushButton.setDisabled(False)
    247.             self.pushButton_2.setText(u"摄像头检测")
    248.  
    249.     def show_video_frame(self):
    250.         name_list = []
    251.  
    252.         flag, img = self.cap.read()
    253.         if img is not None:
    254.             showimg = img
    255.             with torch.no_grad():
    256.                 img = letterbox(img, new_shape=self.opt.img_size)[0]
    257.                 # Convert
    258.                 # BGR to RGB, to 3x416x416
    259.                 img = img[:, :, ::-1].transpose(2, 0, 1)
    260.                 img = np.ascontiguousarray(img)
    261.                 img = torch.from_numpy(img).to(self.device)
    262.                 img = img.half() if self.half else img.float()  # uint8 to fp16/32
    263.                 img /= 255.0  # 0 - 255 to 0.0 - 1.0
    264.                 if img.ndimension() == 3:
    265.                     img = img.unsqueeze(0)
    266.                 # Inference
    267.                 pred = self.model(img, augment=self.opt.augment)[0]
    268.  
    269.                 # Apply NMS
    270.                 pred = non_max_suppression(pred, self.opt.conf_thres, self.opt.iou_thres, classes=self.opt.classes,
    271.                                            agnostic=self.opt.agnostic_nms)
    272.                 # Process detections
    273.                 for i, det in enumerate(pred):  # detections per image
    274.                     if det is not None and len(det):
    275.                         # Rescale boxes from img_size to im0 size
    276.                         det[:, :4] = scale_coords(
    277.                             img.shape[2:], det[:, :4], showimg.shape).round()
    278.                         # Write results
    279.                         for *xyxy, conf, cls in reversed(det):
    280.                             label = '%s %.2f' % (self.names[int(cls)], conf)
    281.                             name_list.append(self.names[int(cls)])
    282.                             print(label)
    283.                             plot_one_box(
    284.                                 xyxy, showimg, label=label, color=self.colors[int(cls)], line_thickness=2)
    285.  
    286.             self.out.write(showimg)
    287.             show = cv2.resize(showimg, (640, 480))
    288.             self.result = cv2.cvtColor(show, cv2.COLOR_BGR2RGB)
    289.             showImage = QtGui.QImage(self.result.data, self.result.shape[1], self.result.shape[0],
    290.                                      QtGui.QImage.Format_RGB888)
    291.             self.label.setPixmap(QtGui.QPixmap.fromImage(showImage))
    292.  
    293.         else:
    294.             self.timer_video.stop()
    295.             self.cap.release()
    296.             self.out.release()
    297.             self.label.clear()
    298.             self.pushButton_3.setDisabled(False)
    299.             self.pushButton.setDisabled(False)
    300.             self.pushButton_2.setDisabled(False)
    301.             self.init_logo()
    302.  
    303.  
    304. if __name__ == '__main__':
    305.     app = QtWidgets.QApplication(sys.argv)
    306.     ui = Ui_MainWindow()
    307.     ui.show()
    308.     sys.exit(app.exec_())

    5.添加背景图片

    将demo中最后一段代码改为如下，其中background-image为背景图片地址。

    1. if __name__ == '__main__':
    2.     stylesheet = """
    3.             Ui_MainWindow {
    4.                 background-image: url("4K.jpg");
    5.                 background-repeat: no-repeat;
    6.                 background-position: center;
    7.             }
    8.         """
    9.     app = QtWidgets.QApplication(sys.argv)
    10.     app.setStyleSheet(stylesheet)
    11.     ui = Ui_MainWindow()
    12.     ui.show()
    13.     sys.exit(app.exec_())

     

    6.reference

    http://t.csdn.cn/ZVtSKicon-default.png?t=M85Bhttp://t.csdn.cn/ZVtSKPyQt5系列教程(三)利用QtDesigner设计UI界面 - 迷途小书童的Note迷途小书童的Note (xugaoxiang.com)icon-default.png?t=M85Bhttps://xugaoxiang.com/2019/12/04/pyqt5-3-qtdesigner/ 

  • 相关阅读:
    Redis学习笔记①基础篇_Redis快速入门
    五、《图解HTTP》报文首部和HTTP缓存
    freeswitch拨打分机号
    分省/市/县最低工资标准-12-2021年&1949-2020全国/省/市/县GDP数据
    我的Vue之旅 10 Gin重写后端、实现页面详情页 Mysql + Golang + Gin
    看守所视频AI行为分析系统
    【无标题】
    华为云新用户云服务器优惠价格表
    springboot+mybatis-plus-join+mysql实现连表查询
    LinkList集合方法（自写）
  • 原文地址：https://blog.csdn.net/m0_56247038/article/details/127898782