YOLOv8 实时屏幕目标检测与透明框选叠加教程

胖静姑娘_2854

胖静姑娘_2854

2026-05-23

721人浏览

原创

本文介绍如何使用 yolov8 结合 pyqt5 在 windows/macos 全屏上实时绘制检测框,无需视频流或跟踪模块,适用于桌面截图场景(如自动化测试、游戏辅助、ui 元素识别等)。

本文介绍如何使用 yolov8 结合 pyqt5 在 windows/macos 全屏上实时绘制检测框,无需视频流或跟踪模块,适用于桌面截图场景(如自动化测试、游戏辅助、ui 元素识别等)。

在基于屏幕截图的目标检测任务中(如监控特定 UI 元素、识别弹窗按钮、捕捉游戏内对象),仅调用 YOLO.predict() 得到结果是不够的——关键在于将预测出的边界框(bounding boxes)实时、无遮挡、高帧率地叠加显示在原始屏幕上。由于 cv2.imshow() 会新建窗口且无法置顶/透明,直接绘图不可行;而 pyautogui.drawRectangle() 性能差、不支持多框、无法持续刷新。因此,轻量级透明覆盖层(Overlay)是最佳实践方案。

以下是一个稳定、可部署的完整实现,基于 PyQt5 构建常驻顶层透明窗口,并通过 paintEvent 动态渲染 YOLOv8 检测结果:

import sys
from PyQt5.QtWidgets import QApplication, QWidget
from PyQt5.QtGui import QPainter, QPen, QColor
from PyQt5.QtCore import Qt, QRect
import pyautogui
import cv2
import numpy as np
from ultralytics import YOLO

class DetectionOverlay(QWidget):
    def __init__(self, model_path: str, conf_threshold: float = 0.4):
        super().__init__()
        self.model = YOLO(model_path)
        self.conf_threshold = conf_threshold
        self.detections = []  # List of [x1, y1, x2, y2]
        self.init_ui()

    def init_ui(self):
        screen_size = pyautogui.size()
        self.setGeometry(0, 0, screen_size.width, screen_size.height)
        self.setWindowTitle("YOLOv8 Screen Overlay")
        self.setWindowFlags(Qt.FramelessWindowHint | Qt.WindowStaysOnTopHint | Qt.Tool)
        self.setAttribute(Qt.WA_TranslucentBackground)
        self.setAttribute(Qt.WA_NoSystemBackground)

    def paintEvent(self, event):
        painter = QPainter(self)
        pen = QPen(QColor(0, 255, 0), 2, Qt.SolidLine)  # Green, 2px
        painter.setPen(pen)
        painter.setRenderHint(QPainter.Antialiasing)

        for (x1, y1, x2, y2) in self.detections:
            # Ensure coordinates are within screen bounds
            x1, y1 = max(0, int(x1)), max(0, int(y1))
            x2, y2 = min(self.width(), int(x2)), min(self.height(), int(y2))
            if x1  0 and hasattr(results[0].boxes, 'xyxy') and len(results[0].boxes.xyxy) > 0:
                boxes = results[0].boxes.xyxy.cpu().numpy()
                overlay.detections = boxes.astype(int).tolist()

            overlay.repaint()
            QApplication.processEvents()
            app.processEvents()

    except KeyboardInterrupt:
        print("\nOverlay stopped.")
    finally:
        sys.exit(app.exec_())

if __name__ == "__main__":
    main()

✅ 关键优化说明:

  • 使用 Qt.Tool | Qt.FramelessWindowHint | Qt.WindowStaysOnTopHint 确保窗口始终置顶且无边框;
  • WA_TranslucentBackground + WA_NoSystemBackground 实现真透明(非黑色背景);
  • QPainter.Antialiasing 提升矩形边缘清晰度;
  • 坐标边界校验防止越界绘图崩溃;
  • 首次推理预热(warm-up)显著降低首帧延迟;
  • conf_threshold 直接传入 .model() 调用,比后过滤更高效。

⚠️ 注意事项:

  • 性能提示:全屏截图(如 2560×1440)+ YOLOv8m/l 推理可能达 50–150ms/帧。若卡顿,可:① 降低截图分辨率(pyautogui.screenshot(region=(...)) 裁剪 ROI);② 改用 YOLO(model).to('cuda') 启用 GPU;③ 设置 time.sleep(0.05) 控制帧率。
  • 跨平台兼容性:本方案在 Windows/macOS 均可运行;Linux 需额外配置窗口管理器(如启用 _NET_WM_STATE_ABOVE)。
  • 安全限制:macOS 可能要求授予「屏幕录制」权限;Windows 用户需关闭「硬件加速」(部分显卡驱动下 PyQt 透明层异常)。
  • 进阶扩展:可添加类别标签(QPainter.drawText())、置信度颜色映射、点击穿透(setAttribute(Qt.WA_TransparentForMouseEvents))或热键退出。

该方案已验证于 YOLOv8.2+、PyQt5 5.15+、Python 3.9+ 环境,兼顾简洁性与生产可用性——无需 OpenCV GUI 窗口干扰,真正实现“所见即检测”。

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

2023.07.20

1671

4

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

2023.07.25

4204

7

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

2023.07.31

1669

3

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

2023.08.03

24417

23

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

2987

5

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.04

3007

5

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

1163

5

python合并两个列表
python合并两个列表

Python是一种强大的编程语言,具有许多方便的功能和工具。在Python中,有多种方法可以合并两个列表。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

2023.08.10

596

4

python是前端还是后端
python是前端还是后端

Python属于前端也属于后端,其灵活性和丰富的生态系统使得开发人员能够在不同的领域中灵活运用。本专题为大家提供python相关的文章、下载、课程内容,供大家免费下载体验。

2023.08.11

2323

5

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
热门推荐
/
最新课程
phpStudy极速入门视频教程
phpStudy极速入门视频教程

共6课时 | 54.6万人学习

独孤九贱(4)_PHP视频教程
独孤九贱(4)_PHP视频教程

共89课时 | 133.4万人学习