YOLO与PyQt5结合-增加论文工作量-实现一个目标检测的UI界面

2024-08-27 23:12

本文主要是介绍YOLO与PyQt5结合-增加论文工作量-实现一个目标检测的UI界面,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

这是个简单的界面,Qtdesigner支持各种界面,支持替换背景添加图标等。

接下来实现一个简单YOLO目标检测界面:

功能:

1、在窗口打开视频或图片进行目标检测,具有中断检测功能:比如检测视频的时候突然打开图片检测,后台就会停止检测,而不是一直执行视频检测。

2、可以输出目标类别,目标中心点,置信度等,此部分为动态生成,每一帧图片开始之前会清空。

步骤:

打开Qt Designer

e76685cfe45f419cb916917a38d440f8.png

保存为ui文件,然后将其转换为py文件。

如下:

# -*- coding: utf-8 -*-# Form implementation generated from reading ui file 'GUi.ui'
#
# Created by: PyQt5 UI code generator 5.15.11
#
# 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, QtWidgetsclass Ui_MainWindow(object):def setupUi(self, MainWindow):MainWindow.setObjectName("MainWindow")MainWindow.resize(1600, 900)MainWindow.setMinimumSize(QtCore.QSize(1600, 900))MainWindow.setMaximumSize(QtCore.QSize(1600, 900))self.centralwidget = QtWidgets.QWidget(MainWindow)self.centralwidget.setObjectName("centralwidget")self.label = QtWidgets.QLabel(self.centralwidget)self.label.setGeometry(QtCore.QRect(20, 10, 851, 471))self.label.setStyleSheet("background-color: rgb(0, 0, 0);")self.label.setObjectName("label")self.pushButton_image = QtWidgets.QPushButton(self.centralwidget)self.pushButton_image.setGeometry(QtCore.QRect(1162, 70, 251, 28))self.pushButton_image.setObjectName("pushButton_image")self.pushButton_video = QtWidgets.QPushButton(self.centralwidget)self.pushButton_video.setGeometry(QtCore.QRect(1162, 130, 251, 28))self.pushButton_video.setObjectName("pushButton_video")self.pushButton_end = QtWidgets.QPushButton(self.centralwidget)self.pushButton_end.setGeometry(QtCore.QRect(1170, 280, 251, 28))self.pushButton_end.setObjectName("pushButton_end")self.tableWidget = QtWidgets.QTableWidget(self.centralwidget)self.tableWidget.setGeometry(QtCore.QRect(20, 521, 400, 300))self.tableWidget.setMinimumSize(QtCore.QSize(400, 300))self.tableWidget.setMaximumSize(QtCore.QSize(400, 300))self.tableWidget.setObjectName("tableWidget")self.tableWidget.setColumnCount(3)self.tableWidget.setRowCount(0)item = QtWidgets.QTableWidgetItem()self.tableWidget.setHorizontalHeaderItem(0, item)item = QtWidgets.QTableWidgetItem()self.tableWidget.setHorizontalHeaderItem(1, item)item = QtWidgets.QTableWidgetItem()self.tableWidget.setHorizontalHeaderItem(2, item)self.line = QtWidgets.QFrame(self.centralwidget)self.line.setGeometry(QtCore.QRect(20, 490, 401, 141))self.line.setFrameShape(QtWidgets.QFrame.HLine)self.line.setFrameShadow(QtWidgets.QFrame.Sunken)self.line.setObjectName("line")MainWindow.setCentralWidget(self.centralwidget)self.statusbar = QtWidgets.QStatusBar(MainWindow)self.statusbar.setObjectName("statusbar")MainWindow.setStatusBar(self.statusbar)self.retranslateUi(MainWindow)QtCore.QMetaObject.connectSlotsByName(MainWindow)def retranslateUi(self, MainWindow):_translate = QtCore.QCoreApplication.translateMainWindow.setWindowTitle(_translate("MainWindow", "MainWindow"))self.label.setText(_translate("MainWindow", "TextLabel"))self.pushButton_image.setText(_translate("MainWindow", "图片"))self.pushButton_video.setText(_translate("MainWindow", "视频"))self.pushButton_end.setText(_translate("MainWindow", "结束检测"))item = self.tableWidget.horizontalHeaderItem(0)item.setText(_translate("MainWindow", "目标名称"))item = self.tableWidget.horizontalHeaderItem(1)item.setText(_translate("MainWindow", "目标中心位置 "))item = self.tableWidget.horizontalHeaderItem(2)item.setText(_translate("MainWindow", "置信度"))

功能实现:

import sysimport numpy as npfrom ultralytics import YOLO
import GUi  
import sys
from PyQt5.QtWidgets import QMessageBox, QFileDialog, QLineEdit
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
from PyQt5.QtCore import *
from PyQt5.QtMultimedia import QMediaPlayer, QMediaContent
import cv2
import sqlite3cla = {0: 'person',1: 'bicycle',2: 'car',3: 'motorcycle',4: 'airplane',5: 'bus',6: 'train',7: 'truck',8: 'boat',9: 'traffic light',10: 'fire hydrant'}class Image_open(QMainWindow, GUi.Ui_MainWindow):def __init__(self, parent=None):super(Image_open, self).__init__(parent)# UI界面self.Video = Falseself.setupUi(self)self.handle_buttons()self.model = YOLO(r'C:\Users\hp\Desktop\ChangeYOLOV\yolov8_pt\yolov8n.pt')def handle_buttons(self):#初始化绑定按钮,在初始化__initi__时调动self.pushButton_image.clicked.connect(self.open_image)self.pushButton_video.clicked.connect(self.display_video)self.pushButton_end.clicked.connect(self.enddect)def enddect(self):#结束检测passdef open_image(self):#图片检测passdef display_video(self):#视频检测passdef closeEvent(self, event):#关闭总窗口之前询问reply = QMessageBox.question(self, '退出', "是否要退出该界面?",QMessageBox.Yes | QMessageBox.No, QMessageBox.No)if reply == QMessageBox.Yes:self.close()event.accept()else:event.ignore()

效果:

1、图片检测

50d2875b3e1a43b097124c5fabbf25df.png

2、视频检测

b65243271243470e90a1b63a626f0f10.png

 

这篇关于YOLO与PyQt5结合-增加论文工作量-实现一个目标检测的UI界面的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!


原文地址:
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.chinasem.cn/article/1113054

相关文章

使用Sentinel自定义返回和实现区分来源方式

《使用Sentinel自定义返回和实现区分来源方式》:本文主要介绍使用Sentinel自定义返回和实现区分来源方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录Sentinel自定义返回和实现区分来源1. 自定义错误返回2. 实现区分来源总结Sentinel自定

Java实现时间与字符串互相转换详解

《Java实现时间与字符串互相转换详解》这篇文章主要为大家详细介绍了Java中实现时间与字符串互相转换的相关方法,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下... 目录一、日期格式化为字符串(一)使用预定义格式(二)自定义格式二、字符串解析为日期(一)解析ISO格式字符串(二)解析自定义

opencv图像处理之指纹验证的实现

《opencv图像处理之指纹验证的实现》本文主要介绍了opencv图像处理之指纹验证的实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学... 目录一、简介二、具体案例实现1. 图像显示函数2. 指纹验证函数3. 主函数4、运行结果三、总结一、

Springboot处理跨域的实现方式(附Demo)

《Springboot处理跨域的实现方式(附Demo)》:本文主要介绍Springboot处理跨域的实现方式(附Demo),具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不... 目录Springboot处理跨域的方式1. 基本知识2. @CrossOrigin3. 全局跨域设置4.

Spring Boot 3.4.3 基于 Spring WebFlux 实现 SSE 功能(代码示例)

《SpringBoot3.4.3基于SpringWebFlux实现SSE功能(代码示例)》SpringBoot3.4.3结合SpringWebFlux实现SSE功能,为实时数据推送提供... 目录1. SSE 简介1.1 什么是 SSE?1.2 SSE 的优点1.3 适用场景2. Spring WebFlu

基于SpringBoot实现文件秒传功能

《基于SpringBoot实现文件秒传功能》在开发Web应用时,文件上传是一个常见需求,然而,当用户需要上传大文件或相同文件多次时,会造成带宽浪费和服务器存储冗余,此时可以使用文件秒传技术通过识别重复... 目录前言文件秒传原理代码实现1. 创建项目基础结构2. 创建上传存储代码3. 创建Result类4.

SpringBoot日志配置SLF4J和Logback的方法实现

《SpringBoot日志配置SLF4J和Logback的方法实现》日志记录是不可或缺的一部分,本文主要介绍了SpringBoot日志配置SLF4J和Logback的方法实现,文中通过示例代码介绍的非... 目录一、前言二、案例一:初识日志三、案例二:使用Lombok输出日志四、案例三:配置Logback一

Python如何使用__slots__实现节省内存和性能优化

《Python如何使用__slots__实现节省内存和性能优化》你有想过,一个小小的__slots__能让你的Python类内存消耗直接减半吗,没错,今天咱们要聊的就是这个让人眼前一亮的技巧,感兴趣的... 目录背景:内存吃得满满的类__slots__:你的内存管理小助手举个大概的例子:看看效果如何?1.

Python+PyQt5实现多屏幕协同播放功能

《Python+PyQt5实现多屏幕协同播放功能》在现代会议展示、数字广告、展览展示等场景中,多屏幕协同播放已成为刚需,下面我们就来看看如何利用Python和PyQt5开发一套功能强大的跨屏播控系统吧... 目录一、项目概述:突破传统播放限制二、核心技术解析2.1 多屏管理机制2.2 播放引擎设计2.3 专

Python实现无痛修改第三方库源码的方法详解

《Python实现无痛修改第三方库源码的方法详解》很多时候,我们下载的第三方库是不会有需求不满足的情况,但也有极少的情况,第三方库没有兼顾到需求,本文将介绍几个修改源码的操作,大家可以根据需求进行选择... 目录需求不符合模拟示例 1. 修改源文件2. 继承修改3. 猴子补丁4. 追踪局部变量需求不符合很