yolo小damo合集

news2025/1/7 4:42:48

 效果如下:这个是图片检测

 效果如下:这个是视频检测 

 效果如下:这个是摄像头检测 

1 相关库

除了yolov11所用库之外,本文所用到的额外库为pyqt5,输入指令进行安装

pip install pyqt5
导入所需要的库
import sys
from PyQt5.QtCore import Qt, QTimer
from PyQt5.QtWidgets import QApplication, QMainWindow, QLabel, QVBoxLayout, QWidget, QPushButton, QHBoxLayout, QMessageBox, QFileDialog
from PyQt5.QtGui import QImage, QPixmap, QIcon
import cv2
from ultralytics import YOLO

2 界面

使用pyqt5进行界面创作

"""
        初始化主窗口。
        这个方法设置了窗口的标题、图标、大小,并初始化了各种组件,如标签、按钮、布局等。
        还创建了用于摄像头检测的定时器,并设置了初始状态为未暂停。
        """
        super().__init__()
        # 设置窗口标题
        self.setWindowTitle("@author:晓晓")
        # self.setWindowIcon(QIcon("icon.png"))
        # 设置窗口大小
        self.setGeometry(300, 150, 800, 400)

        # 创建两个QLabel分别显示左右图像
        # 左侧图像标签,用于显示原始图像
        self.label1 = QLabel()
        # 设置标签内容居中对齐
        self.label1.setAlignment(Qt.AlignCenter)
        # 设置标签最小尺寸
        self.label1.setMinimumSize(580, 450)  
        # 设置标签样式,添加边框并设置背景颜色为黑色
        self.label1.setStyleSheet('border:3px solid #6950a1; background-color: black;')  

        # 右侧图像标签,用于显示检测后的图像
        self.label2 = QLabel()
        # 设置标签内容居中对齐
        self.label2.setAlignment(Qt.AlignCenter)
        # 设置标签最小尺寸
        self.label2.setMinimumSize(580, 450)  
        # 设置标签样式,添加边框并设置背景颜色为黑色
        self.label2.setStyleSheet('border:3px solid #6950a1; background-color: black;')  

        ###摄像头检测
        # 用于存储摄像头对象
        self.camera = None  
        # 定时器,用于定时从摄像头获取帧
        self.camera_timer = QTimer()  
        # 定时器超时连接到更新摄像头帧的方法
        self.camera_timer.timeout.connect(self.update_camera_frame)  

        # 新增属性,用于标记视频检测是否暂停
        self.is_video_paused = False
        # 新增属性,用于标记摄像头检测是否暂停
        self.is_camera_paused = False

        # 水平布局,用于放置左右两个QLabel
        layout = QVBoxLayout()
        # layout.addWidget(self.label1)
        hbox_video = QHBoxLayout()
        # 在水平布局中添加左侧标签,用于显示原始图像
        hbox_video.addWidget(self.label1)  
        # 在水平布局中添加右侧标签,用于显示检测后的图像
        hbox_video.addWidget(self.label2)  
        # 将水平布局添加到垂直布局中
        layout.addLayout(hbox_video)
        # 创建Worker对象
        self.worker = Worker()
        # 创建按钮布局
        hbox_buttons = QHBoxLayout()
        # 添加模型选择按钮
        self.load_model_button = QPushButton("📁模型选择")
        # 点击按钮时连接到load_model方法
        self.load_model_button.clicked.connect(self.load_model)
        # 设置按钮固定大小
        self.load_model_button.setFixedSize(120, 30)
        # 将按钮添加到按钮布局中
        hbox_buttons.addWidget(self.load_model_button)

        # 添加图片检测按钮
        self.image_detect_button = QPushButton("💾图片检测")
        # 点击按钮时连接到detect_image方法
        self.image_detect_button.clicked.connect(self.detect_image)
        # 初始状态下禁用按钮
        self.image_detect_button.setEnabled(False)
        # 设置按钮固定大小
        self.image_detect_button.setFixedSize(120, 30)
        # 将按钮添加到按钮布局中
        hbox_buttons.addWidget(self.image_detect_button)

        # 添加视频检测按钮
        self.video_detect_button = QPushButton("🎬视频检测")
        # 点击按钮时连接到detect_video_start方法
        self.video_detect_button.clicked.connect(self.detect_video_start)
        # 初始状态下禁用按钮
        self.video_detect_button.setEnabled(False)
        # 设置按钮固定大小
        self.video_detect_button.setFixedSize(120, 30)
        # 将按钮添加到按钮布局中
        hbox_buttons.addWidget(self.video_detect_button)

        # 添加摄像头检测按钮
        self.camera_detect_button = QPushButton("📷摄像头检测")
        # 点击按钮时连接到start_camera_detection方法
        self.camera_detect_button.clicked.connect(self.start_camera_detection)
        # 初始状态下禁用按钮
        self.camera_detect_button.setEnabled(False)
        # 设置按钮固定大小
        self.camera_detect_button.setFixedSize(120, 30)
        # 将按钮添加到按钮布局中
        hbox_buttons.addWidget(self.camera_detect_button)

        # 添加显示检测物体按钮
        self.display_objects_button = QPushButton("🔍显示检测物体")
        # 点击按钮时连接到show_detected_objects方法
        self.display_objects_button.clicked.connect(self.show_detected_objects)
        # 初始状态下禁用按钮
        self.display_objects_button.setEnabled(False)
        # 设置按钮固定大小
        self.display_objects_button.setFixedSize(120, 30)
        # 将按钮添加到按钮布局中
        hbox_buttons.addWidget(self.display_objects_button)

        # 添加退出按钮
        self.exit_button = QPushButton("❌退出")
        # 点击按钮时连接到exit_application方法
        self.exit_button.clicked.connect(self.exit_application)
        # 设置按钮固定大小
        self.exit_button.setFixedSize(120, 30)
        # 将按钮添加到按钮布局中
        hbox_buttons.addWidget(self.exit_button)

        # 将按钮布局添加到垂直布局中
        layout.addLayout(hbox_buttons)
        # 创建中心部件
        central_widget = QWidget()
        # 设置中心部件的布局
        central_widget.setLayout(layout)
        # 设置窗口的中心部件
        self.setCentralWidget(central_widget)

        # 用于存储当前检测结果
        self.current_results = None

3实现

每个部分的按钮实现

    def detect_image(self):
        if self.camera_timer.isActive():  # 判断摄像头检测是否正在运行,如果是则关闭
            self.camera_timer.stop()
            if self.camera is not None:
                self.camera.release()

        image_path, _ = QFileDialog.getOpenFileName(None, "选择图片文件", "", "图片文件 (*.jpg *.jpeg *.png)")
        if image_path:
            image = cv2.imread(image_path)
            if image is not None:
                self.current_results = self.worker.model.predict(image)
                if self.current_results:
                    annotated_image = self.current_results[0].plot()
                    image_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)  # 转换为RGB
                    height1, width1, channel1 = image_rgb.shape
                    bytesPerLine1 = 3 * width1
                    qimage1 = QImage(image_rgb.data, width1, height1, bytesPerLine1, QImage.Format_RGB888)
                    pixmap1 = QPixmap.fromImage(qimage1)
                    self.label1.setPixmap(pixmap1.scaled(self.label1.size(), Qt.KeepAspectRatio))

                    annotated_image = cv2.cvtColor(annotated_image, cv2.COLOR_BGR2RGB)  # 转换为RGB
                    height2, width2, channel2 = annotated_image.shape
                    bytesPerLine2 = 3 * width2
                    qimage2 = QImage(annotated_image.data, width2, height2, bytesPerLine2, QImage.Format_RGB888)
                    pixmap2 = QPixmap.fromImage(qimage2)
                    self.label2.setPixmap(pixmap2.scaled(self.label2.size(), Qt.KeepAspectRatio))

    def detect_video_start(self):
        if self.camera_timer.isActive():  # 判断摄像头检测是否正在运行,如果是则关闭
            self.camera_timer.stop()
            if self.camera is not None:
                self.camera.release()

        video_path, _ = QFileDialog.getOpenFileName(None, "选择视频文件", "", "视频文件 (*.mp4 *.avi)")
        if video_path:
            cap = cv2.VideoCapture(video_path)
            while cap.isOpened():
                success, frame = cap.read()
                if success:
                    frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
                    height1, width1, channel1 = frame_rgb.shape
                    bytesPerLine1 = 3 * width1
                    qimage = QImage(frame_rgb.data, width1, height1, bytesPerLine1, QImage.Format_RGB888)
                    pixmap = QPixmap.fromImage(qimage)
                    self.label1.setPixmap(pixmap.scaled(self.label1.size(), Qt.KeepAspectRatio))

                    results = self.worker.model.predict(frame)
                    # Visualize the results on the frame
                    annotated_frame = results[0].plot()

                    frame_rgb = cv2.cvtColor(annotated_frame, cv2.COLOR_BGR2RGB)
                    height2, width2, channel2 = frame_rgb.shape
                    bytesPerLine2 = 3 * width2
                    qimage = QImage(frame_rgb.data, width2, height2, bytesPerLine2, QImage.Format_RGB888)
                    pixmap = QPixmap.fromImage(qimage)
                    self.label2.setPixmap(pixmap.scaled(self.label2.size(), Qt.KeepAspectRatio))

                    if cv2.waitKey(1) & 0xFF == ord("q"):
                        break
                else:
                    # Break the loop if the end of the video is reached
                    break

            # Release the video capture object and close the display window
            cap.release()
            cv2.destroyAllWindows()

    def start_camera_detection(self):
        self.camera = cv2.VideoCapture(0)  # 打开默认摄像头(索引为0)
        if self.camera.isOpened():
            self.camera_timer.start(30)  # 设置定时器间隔,每30毫秒更新一帧
        else:
            QMessageBox.warning(self, "摄像头错误", "无法打开摄像头,请检查设备是否连接正常。")

    def update_camera_frame(self):
        success, frame = self.camera.read()
        if success:
            # 对获取到的摄像头帧进行水平翻转,实现镜像效果
            frame = cv2.flip(frame, 1)
            frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
            height1, width1, channel1 = frame_rgb.shape
            bytesPerLine1 = 3 * width1
            qimage = QImage(frame_rgb.data, width1, height1, bytesPerLine1, QImage.Format_RGB888)
            pixmap = QPixmap.fromImage(qimage)
            self.label1.setPixmap(pixmap.scaled(self.label1.size(), Qt.KeepAspectRatio))

            results = self.worker.model.predict(frame)
            # 此处修改,将每次摄像头检测的结果都更新到self.current_results中(以列表形式存储每次结果)
            if self.current_results is None:
                self.current_results = []
            self.current_results.clear()
            self.current_results.append(results)

            annotated_frame = results[0].plot()
            frame_rgb = cv2.cvtColor(annotated_frame, cv2.COLOR_BGR2RGB)
            height2, width2, channel2 = frame_rgb.shape
            bytesPerLine2 = 3 * width2
            qimage = QImage(frame_rgb.data, width2, height2, bytesPerLine2, QImage.Format_RGB888)
            pixmap = QPixmap.fromImage(qimage)
            self.label2.setPixmap(pixmap.scaled(self.label2.size(), Qt.KeepAspectRatio))
        else:
            self.camera_timer.stop()
            self.camera.release()
            QMessageBox.warning(self, "摄像头错误", "摄像头读取失败,请检查设备是否正常工作。")

    def show_detected_objects(self):
        if self.current_results:
            all_det_info = []
            for result in self.current_results:
                det_info = result[0].boxes.cls
                all_det_info.extend(det_info.tolist())
            object_count = len(all_det_info)
            object_info = f"识别到的物体总个数:{object_count}\n"
            object_dict = {}
            class_names_dict = self.current_results[0][0].names
            for class_id in all_det_info:
                class_name = class_names_dict[int(class_id)]
                if class_name in object_dict:
                    object_dict[class_name] += 1
                else:
                    object_dict[class_name] = 1
            sorted_objects = sorted(object_dict.items(), key=lambda x: x[1], reverse=True)
            for obj_name, obj_count in sorted_objects:
                object_info += f"{obj_name}: {obj_count}\n"
            if self.video_detect_button.isEnabled() and self.video_detect_button.isChecked():
                self.is_video_paused = True
            elif self.camera_detect_button.isEnabled() and self.camera_detect_button.isChecked():
                self.is_camera_paused = True

            msg_box = QMessageBox(self)
            msg_box.setWindowTitle("识别结果")
            msg_box.setText(object_info)
            msg_box.finished.connect(self.resume_detection)
            msg_box.exec_()
        else:
            self.show_message_box("识别结果", "未检测到物体")

    def show_message_box(self, title, message):
        msg_box = QMessageBox(self)
        msg_box.setWindowTitle(title)
        msg_box.setText(message)
        msg_box.exec_()

    def load_model(self):
        if self.worker.load_model():
            self.image_detect_button.setEnabled(True)
            self.display_objects_button.setEnabled(True)
            self.video_detect_button.setEnabled(True)
            self.camera_detect_button.setEnabled(True)

    def resume_detection(self):
        if self.is_video_paused:
            self.is_video_paused = False
        elif self.is_camera_paused:
            self.is_camera_paused = False

    def exit_application(self):
        # 终止程序运行
        sys.exit()

4总代码

import sys
from PyQt5.QtCore import Qt, QTimer
from PyQt5.QtWidgets import QApplication, QMainWindow, QLabel, QVBoxLayout, QWidget, QPushButton, QHBoxLayout, QMessageBox, QFileDialog
from PyQt5.QtGui import QImage, QPixmap, QIcon
import cv2
from ultralytics import YOLO


class Worker:
    def __init__(self):
        self.model = None

    def load_model(self):
        model_path, _ = QFileDialog.getOpenFileName(None, "选择模型文件", "", "模型文件 (*.pt)")
        if model_path:
            self.model = YOLO(model_path)
            return self.model is not None
        return False


class MainWindow(QMainWindow):
    def __init__(self):
        """
        初始化主窗口。
        这个方法设置了窗口的标题、图标、大小,并初始化了各种组件,如标签、按钮、布局等。
        还创建了用于摄像头检测的定时器,并设置了初始状态为未暂停。
        """
        super().__init__()
        # 设置窗口标题
        self.setWindowTitle("@author:晓晓")
        # self.setWindowIcon(QIcon("icon.png"))
        # 设置窗口大小
        self.setGeometry(300, 150, 800, 400)

        # 创建两个QLabel分别显示左右图像
        # 左侧图像标签,用于显示原始图像
        self.label1 = QLabel()
        # 设置标签内容居中对齐
        self.label1.setAlignment(Qt.AlignCenter)
        # 设置标签最小尺寸
        self.label1.setMinimumSize(580, 450)  
        # 设置标签样式,添加边框并设置背景颜色为黑色
        self.label1.setStyleSheet('border:3px solid #6950a1; background-color: black;')  

        # 右侧图像标签,用于显示检测后的图像
        self.label2 = QLabel()
        # 设置标签内容居中对齐
        self.label2.setAlignment(Qt.AlignCenter)
        # 设置标签最小尺寸
        self.label2.setMinimumSize(580, 450)  
        # 设置标签样式,添加边框并设置背景颜色为黑色
        self.label2.setStyleSheet('border:3px solid #6950a1; background-color: black;')  

        ###摄像头检测
        # 用于存储摄像头对象
        self.camera = None  
        # 定时器,用于定时从摄像头获取帧
        self.camera_timer = QTimer()  
        # 定时器超时连接到更新摄像头帧的方法
        self.camera_timer.timeout.connect(self.update_camera_frame)  

        # 新增属性,用于标记视频检测是否暂停
        self.is_video_paused = False
        # 新增属性,用于标记摄像头检测是否暂停
        self.is_camera_paused = False

        # 水平布局,用于放置左右两个QLabel
        layout = QVBoxLayout()
        # layout.addWidget(self.label1)
        hbox_video = QHBoxLayout()
        # 在水平布局中添加左侧标签,用于显示原始图像
        hbox_video.addWidget(self.label1)  
        # 在水平布局中添加右侧标签,用于显示检测后的图像
        hbox_video.addWidget(self.label2)  
        # 将水平布局添加到垂直布局中
        layout.addLayout(hbox_video)
        # 创建Worker对象
        self.worker = Worker()
        # 创建按钮布局
        hbox_buttons = QHBoxLayout()
        # 添加模型选择按钮
        self.load_model_button = QPushButton("📁模型选择")
        # 点击按钮时连接到load_model方法
        self.load_model_button.clicked.connect(self.load_model)
        # 设置按钮固定大小
        self.load_model_button.setFixedSize(120, 30)
        # 将按钮添加到按钮布局中
        hbox_buttons.addWidget(self.load_model_button)

        # 添加图片检测按钮
        self.image_detect_button = QPushButton("💾图片检测")
        # 点击按钮时连接到detect_image方法
        self.image_detect_button.clicked.connect(self.detect_image)
        # 初始状态下禁用按钮
        self.image_detect_button.setEnabled(False)
        # 设置按钮固定大小
        self.image_detect_button.setFixedSize(120, 30)
        # 将按钮添加到按钮布局中
        hbox_buttons.addWidget(self.image_detect_button)

        # 添加视频检测按钮
        self.video_detect_button = QPushButton("🎬视频检测")
        # 点击按钮时连接到detect_video_start方法
        self.video_detect_button.clicked.connect(self.detect_video_start)
        # 初始状态下禁用按钮
        self.video_detect_button.setEnabled(False)
        # 设置按钮固定大小
        self.video_detect_button.setFixedSize(120, 30)
        # 将按钮添加到按钮布局中
        hbox_buttons.addWidget(self.video_detect_button)

        # 添加摄像头检测按钮
        self.camera_detect_button = QPushButton("📷摄像头检测")
        # 点击按钮时连接到start_camera_detection方法
        self.camera_detect_button.clicked.connect(self.start_camera_detection)
        # 初始状态下禁用按钮
        self.camera_detect_button.setEnabled(False)
        # 设置按钮固定大小
        self.camera_detect_button.setFixedSize(120, 30)
        # 将按钮添加到按钮布局中
        hbox_buttons.addWidget(self.camera_detect_button)

        # 添加显示检测物体按钮
        self.display_objects_button = QPushButton("🔍显示检测物体")
        # 点击按钮时连接到show_detected_objects方法
        self.display_objects_button.clicked.connect(self.show_detected_objects)
        # 初始状态下禁用按钮
        self.display_objects_button.setEnabled(False)
        # 设置按钮固定大小
        self.display_objects_button.setFixedSize(120, 30)
        # 将按钮添加到按钮布局中
        hbox_buttons.addWidget(self.display_objects_button)

        # 添加退出按钮
        self.exit_button = QPushButton("❌退出")
        # 点击按钮时连接到exit_application方法
        self.exit_button.clicked.connect(self.exit_application)
        # 设置按钮固定大小
        self.exit_button.setFixedSize(120, 30)
        # 将按钮添加到按钮布局中
        hbox_buttons.addWidget(self.exit_button)

        # 将按钮布局添加到垂直布局中
        layout.addLayout(hbox_buttons)
        # 创建中心部件
        central_widget = QWidget()
        # 设置中心部件的布局
        central_widget.setLayout(layout)
        # 设置窗口的中心部件
        self.setCentralWidget(central_widget)

        # 用于存储当前检测结果
        self.current_results = None

    def detect_image(self):
        if self.camera_timer.isActive():  # 判断摄像头检测是否正在运行,如果是则关闭
            self.camera_timer.stop()
            if self.camera is not None:
                self.camera.release()

        image_path, _ = QFileDialog.getOpenFileName(None, "选择图片文件", "", "图片文件 (*.jpg *.jpeg *.png)")
        if image_path:
            image = cv2.imread(image_path)
            if image is not None:
                self.current_results = self.worker.model.predict(image)
                if self.current_results:
                    annotated_image = self.current_results[0].plot()
                    image_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)  # 转换为RGB
                    height1, width1, channel1 = image_rgb.shape
                    bytesPerLine1 = 3 * width1
                    qimage1 = QImage(image_rgb.data, width1, height1, bytesPerLine1, QImage.Format_RGB888)
                    pixmap1 = QPixmap.fromImage(qimage1)
                    self.label1.setPixmap(pixmap1.scaled(self.label1.size(), Qt.KeepAspectRatio))

                    annotated_image = cv2.cvtColor(annotated_image, cv2.COLOR_BGR2RGB)  # 转换为RGB
                    height2, width2, channel2 = annotated_image.shape
                    bytesPerLine2 = 3 * width2
                    qimage2 = QImage(annotated_image.data, width2, height2, bytesPerLine2, QImage.Format_RGB888)
                    pixmap2 = QPixmap.fromImage(qimage2)
                    self.label2.setPixmap(pixmap2.scaled(self.label2.size(), Qt.KeepAspectRatio))

    def detect_video_start(self):
        if self.camera_timer.isActive():  # 判断摄像头检测是否正在运行,如果是则关闭
            self.camera_timer.stop()
            if self.camera is not None:
                self.camera.release()

        video_path, _ = QFileDialog.getOpenFileName(None, "选择视频文件", "", "视频文件 (*.mp4 *.avi)")
        if video_path:
            cap = cv2.VideoCapture(video_path)
            while cap.isOpened():
                success, frame = cap.read()
                if success:
                    frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
                    height1, width1, channel1 = frame_rgb.shape
                    bytesPerLine1 = 3 * width1
                    qimage = QImage(frame_rgb.data, width1, height1, bytesPerLine1, QImage.Format_RGB888)
                    pixmap = QPixmap.fromImage(qimage)
                    self.label1.setPixmap(pixmap.scaled(self.label1.size(), Qt.KeepAspectRatio))

                    results = self.worker.model.predict(frame)
                    # Visualize the results on the frame
                    annotated_frame = results[0].plot()

                    frame_rgb = cv2.cvtColor(annotated_frame, cv2.COLOR_BGR2RGB)
                    height2, width2, channel2 = frame_rgb.shape
                    bytesPerLine2 = 3 * width2
                    qimage = QImage(frame_rgb.data, width2, height2, bytesPerLine2, QImage.Format_RGB888)
                    pixmap = QPixmap.fromImage(qimage)
                    self.label2.setPixmap(pixmap.scaled(self.label2.size(), Qt.KeepAspectRatio))

                    if cv2.waitKey(1) & 0xFF == ord("q"):
                        break
                else:
                    # Break the loop if the end of the video is reached
                    break

            # Release the video capture object and close the display window
            cap.release()
            cv2.destroyAllWindows()

    def start_camera_detection(self):
        self.camera = cv2.VideoCapture(0)  # 打开默认摄像头(索引为0)
        if self.camera.isOpened():
            self.camera_timer.start(30)  # 设置定时器间隔,每30毫秒更新一帧
        else:
            QMessageBox.warning(self, "摄像头错误", "无法打开摄像头,请检查设备是否连接正常。")

    def update_camera_frame(self):
        success, frame = self.camera.read()
        if success:
            # 对获取到的摄像头帧进行水平翻转,实现镜像效果
            frame = cv2.flip(frame, 1)
            frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
            height1, width1, channel1 = frame_rgb.shape
            bytesPerLine1 = 3 * width1
            qimage = QImage(frame_rgb.data, width1, height1, bytesPerLine1, QImage.Format_RGB888)
            pixmap = QPixmap.fromImage(qimage)
            self.label1.setPixmap(pixmap.scaled(self.label1.size(), Qt.KeepAspectRatio))

            results = self.worker.model.predict(frame)
            # 此处修改,将每次摄像头检测的结果都更新到self.current_results中(以列表形式存储每次结果)
            if self.current_results is None:
                self.current_results = []
            self.current_results.clear()
            self.current_results.append(results)

            annotated_frame = results[0].plot()
            frame_rgb = cv2.cvtColor(annotated_frame, cv2.COLOR_BGR2RGB)
            height2, width2, channel2 = frame_rgb.shape
            bytesPerLine2 = 3 * width2
            qimage = QImage(frame_rgb.data, width2, height2, bytesPerLine2, QImage.Format_RGB888)
            pixmap = QPixmap.fromImage(qimage)
            self.label2.setPixmap(pixmap.scaled(self.label2.size(), Qt.KeepAspectRatio))
        else:
            self.camera_timer.stop()
            self.camera.release()
            QMessageBox.warning(self, "摄像头错误", "摄像头读取失败,请检查设备是否正常工作。")

    def show_detected_objects(self):
        if self.current_results:
            all_det_info = []
            for result in self.current_results:
                det_info = result[0].boxes.cls
                all_det_info.extend(det_info.tolist())
            object_count = len(all_det_info)
            object_info = f"识别到的物体总个数:{object_count}\n"
            object_dict = {}
            class_names_dict = self.current_results[0][0].names
            for class_id in all_det_info:
                class_name = class_names_dict[int(class_id)]
                if class_name in object_dict:
                    object_dict[class_name] += 1
                else:
                    object_dict[class_name] = 1
            sorted_objects = sorted(object_dict.items(), key=lambda x: x[1], reverse=True)
            for obj_name, obj_count in sorted_objects:
                object_info += f"{obj_name}: {obj_count}\n"
            if self.video_detect_button.isEnabled() and self.video_detect_button.isChecked():
                self.is_video_paused = True
            elif self.camera_detect_button.isEnabled() and self.camera_detect_button.isChecked():
                self.is_camera_paused = True

            msg_box = QMessageBox(self)
            msg_box.setWindowTitle("识别结果")
            msg_box.setText(object_info)
            msg_box.finished.connect(self.resume_detection)
            msg_box.exec_()
        else:
            self.show_message_box("识别结果", "未检测到物体")

    def show_message_box(self, title, message):
        msg_box = QMessageBox(self)
        msg_box.setWindowTitle(title)
        msg_box.setText(message)
        msg_box.exec_()

    def load_model(self):
        if self.worker.load_model():
            self.image_detect_button.setEnabled(True)
            self.display_objects_button.setEnabled(True)
            self.video_detect_button.setEnabled(True)
            self.camera_detect_button.setEnabled(True)

    def resume_detection(self):
        if self.is_video_paused:
            self.is_video_paused = False
        elif self.is_camera_paused:
            self.is_camera_paused = False

    def exit_application(self):
        # 终止程序运行
        sys.exit()


if __name__ == '__main__':
    app = QApplication(sys.argv)
    window = MainWindow()
    window.show()
    sys.exit(app.exec_())

喜欢的请三连哦 

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

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

商标名称仅由常见姓氏构成,缺显驳回!

近日一个江苏网友给普推知产商标老杨发过来的一个商标驳回案例,商标驳回的原因与第一次驳回引证的商标居然是不一样的,引证的商标与第一次引证的商标也是不一样的。 看了下引证的两个商标与申请商标名称明显不太近似,或许还有做复审的机会&am…

Rockect基于Dledger的Broker主从同步原理

1.前言 此文章是在儒猿课程中的学习笔记,感兴趣的想看原来的课程可以去咨询儒猿课堂 这篇文章紧挨着上一篇博客来进行编写,有些不清楚的可以看下上一篇博客: RocketMQ原理简述(二)-CSDN博客 2.Broker的高可用 如果…

深入Android架构(从线程到AIDL)_08 认识Android的主线程

目录 3、 认识Android的主线程(又称UI线程) 复习: 各进程(Process)里的主线程​编辑 UI线程的责任: 迅速处理UI事件 举例 3、 认识Android的主线程(又称UI线程) 复习: 各进程(Process)里的主线程 UI线程的责任: 迅速处理UI事…

个人博客自我介绍

你好,我是Chiawei! 大家好,我是Chiawei,一个热爱编程和探索新知识的人。很高兴能在这里与大家分享我的编程之旅。今天,我想和大家聊聊我的自我介绍、编程目标、学习计划以及一些个人想法。 自我介绍 我是一个对技术充…

logback之自定义过滤器

logback有两种过滤器,一种是context中的过滤器叫TurboFilter,是一个全局的过滤器,会影响所有的日志记录。另一种是Appender中的过滤器,只对所在的append有效。两者大同小异,这里我们以Appender的过滤器为例。 &#x…

AcWing练习题:面积

给定三个浮点数 A,B 和 C。 然后,计算如下图形的面积: 底边为 A,高为 C 的三角形。半径 C 的圆。(π3.14159)底边为 A 和 B,高为 C 的梯形。边长为 B 的正方形。边长为 A 和 B 的长方形。 输…

三甲医院等级评审八维数据分析应用(一)--组织、制度、管理可视化篇

一、引言 1.1 研究背景与意义 在当今医疗领域,三甲医院作为医疗服务的核心载体,肩负着保障民众健康、推动医学进步的重任。随着信息技术的飞速发展,数据已成为医院运营管理、医疗质量提升以及科学决策的关键要素。三甲医院等级评审作为衡量医院综合实力与服务水平的重要标…

Spring Bean的初始化过程是怎么样的?

文章内容收录到个人网站,方便阅读:http://hardyfish.top/ 文章内容收录到个人网站,方便阅读:http://hardyfish.top/ 文章内容收录到个人网站,方便阅读:http://hardyfish.top/ Spring Bean 的初始化过程主…

快速上手LangChain(四)LangChain Hub和LangSmith

文章目录 快速上手LangChain(四)LangChain Hub和LangSmith什么是LangChain HubLangChain Hub功能 LangSmith使用 快速上手LangChain(四)LangChain Hub和LangSmith 什么是LangChain Hub LangChain Hub官网地址:https:…

无人机遥控器频率解析!

1.4GHz无人机遥控器工作核心技术 传输距离:1.4GHz频率的无人机遥控器具有较远的传输距离,这使得无人机可以在较大范围内与地面控制站或其他通信节点进行可靠的通信。这种特性特别适用于需要覆盖较大区域的任务或远距离飞行的场景。 穿透力:…

TypyScript从入门到精通

TypyScript从入门到精通 TypyScript 是什么?增加了什么环境搭建二、为何需要 TypeScript三、编译 TypeScript四、类型声明五、类型推断基本类型六、类型总览JavaScript 中的数据类型TypeScript 中的数据类型1. 上述所有 JavaScript 类型2. 六个新类型:3.…

RabbitMq的Java项目实践

在现代软件开发中,消息队列(Message Queue,简称MQ)作为一种重要的组件,承担着上下游消息传递和通信的重任。RabbitMQ作为一款流行的开源消息队列中间件,凭借其高可用性、可扩展性和易用性等特点&#xff0c…

CK40N或CK24标准成本发布出现标记不到物料的情况:评估视图中允许的成本核算变式选择错误

用户打电话反馈标准成本发布在标记环节出现错误: 检查: 1、MM03已经维护了税价1。 2、CK13N: 没有成本没本滚算。 用户是用CK40N处理的, 我让他用CK11N/CK24再处理一次。 仍然报错. 我也试了一个也是报错。 准备CRK1删除原来的…

【HENU】河南大学计院2024 计算机体系结构 期末复习知识点

和光同尘_我的个人主页 一直游到海水变蓝。 体系结构 第一章:计算机系统基础知识计算机系统的实质计算机系统的设计的4个定量原理Amdahl定律CPU性能公式程序的局部性原理: 第二章:指令系统的设计指令系统结构的分类通用寄存器型结构 哈夫曼编码MIPS指令…

Docker 远程访问完整配置教程以及核心参数理解

Docker 远程访问完整配置教程 以下是配置 Docker 支持远程访问的完整教程,包括参数说明、配置修改、云服务器安全组设置、主机防火墙配置,以及验证远程访问的详细步骤。 1. 理解 -H fd:// 参数的作用(理解了以后容易理解后面的操作&#xff…

如何在不丢失数据的情况下从 IOS 14 回滚到 IOS 13

您是否后悔在 iPhone、iPad 或 iPod touch 上安装 iOS 14?如果你这样做,你并不孤单。许多升级到 iOS 14 beta 的 iPhone、iPad 和 iPod touch 用户不再适应它。 如果您在正式发布日期之前升级到 iOS 14 以享受其功能,但您不再适应 iOS 14&am…

vue3 css实现文字输出带光标显示,文字输出完毕,光标消失的效果

Vue实现过程如下&#xff1a; <template><div ><p ref"dom_element" class"typing" :class"{over_fill: record_input_over}"></p></div> </template> <script setup> import {onMounted, ref} from…

Postman测试big-event

报错500。看弹幕&#xff0c;知道可能是yml或sql有问题。 所以检查idea工作台&#xff0c; 直接找UserMapper检查&#xff0c;发现完全OK。 顺着这个error发现可能是sql有问题。因为提示是sql问题&#xff0c;而且是有now()的那个sql。 之后通过给的课件&#xff0c;复制课件…

Wonder Dynamics技术浅析(八):实拍与虚拟合成

该模块旨在将实拍视频与计算机生成的虚拟场景无缝融合&#xff0c;生成具有高度真实感的视觉效果。 一、实拍视频与虚拟场景合成概述 实拍视频与虚拟场景合成的主要目标是将实拍视频中的元素与计算机生成的虚拟场景进行无缝融合&#xff0c;生成具有高度真实感的最终影像。 …

Javascript算法——回溯算法(子集和全排列问题)

子集问题 思路 如果把 子集问题、组合问题、分割问题都抽象为一棵树的话&#xff0c;那么组合问题和分割问题都是收集树的叶子节点&#xff0c;而子集问题是找树的所有节点&#xff01; 78.子集 相比组合问题&#xff0c;此子集问题题目更为简单&#xff0c;收集的是树的所有…