pyqt5无法显示opencv绘制文本和掩码信息

背景:pyqt5无法显示opencv绘制的标签和mask;我们在使用YOLO做实例分割做推理时,会使用opencv做后处理结果绘制(含标签绘制和掩码绘制);结果opencv绘制的解码却无法在pyqt的解码上面显示。pyqt转换代码如下:

python 复制代码
    def convert_frame_to_pixmap(self, frame):
        """将OpenCV帧转换为QPixmap,特别优化mask显示"""
        if frame is None:
            return QPixmap()

        # 转换颜色空间(OpenCV的BGR转Qt的RGB)
        frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)

        # 确保数据类型正确
        if frame_rgb.dtype != np.uint8:
            frame_rgb = cv2.normalize(
                frame_rgb, None, 0, 255, cv2.NORM_MINMAX, dtype=cv2.CV_8U
            )

        # 确保内存连续性
        if not frame_rgb.flags.contiguous:
            frame_rgb = np.ascontiguousarray(frame_rgb)

        h, w, c = frame_rgb.shape

        # 创建QImage - 使用更可靠的参数
        qimg = QImage(
            frame_rgb.data,
            w,
            h,
            frame_rgb.strides[0],  # 直接使用数组的步长
            QImage.Format_RGB888
        )

        # 确保图像不被修改时共享数据,提高性能
        qimg = qimg.copy()

        # 转换为QPixmap并按比例缩放
        pixmap = QPixmap.fromImage(qimg)
        return pixmap.scaled(
            self.display_label.width(),
            self.display_label.height(),
            Qt.KeepAspectRatio,
            Qt.SmoothTransformation
        )

解决: 在使用opencv的imshow可视化看到标签和掩码的结果后,把问题定位在转换和传输。

如果要在qyqt上面显示,首先需要通过pyqt的yqtSignal函数将opencv处理后的帧数据进行传输,然后再将其转换为pyqt的QPixmap。

在转换阶段排查了很久,后来发现不是这个问题。然后问题就定位到传输中和传输前。传输中也不可能,因为能够显示图像,只是无法显示标签和掩码。

来到转换前,这部分的实现是通过处理函数先处理,再将处理后的帧数据传给pyqt的信号槽。最开始是这样实现的:

python 复制代码
self.process_frame(frame)
self.update_frame_signal.emit(frame)
self.process_finished_signal.emit()

打断点测试,发现问题出在这,因为frame数据的标签和掩码在传给信号槽之前会丢失。于是改变写法,换成下面这种更加稳健的写法,就把这个问题解决了。当然这是我代码实现的疏忽。

python 复制代码
frame, results = self.process_frame(frame)
self.update_frame_signal.emit(frame)
self.process_finished_signal.emit()
相关推荐
程序员皮皮林27 分钟前
Java 25 正式发布:更简洁、更高效、更现代!
java·开发语言·python
好家伙VCC29 分钟前
**发散创新:AI绘画编程探索与实践**随着人工智能技术的飞速发展,AI绘
java·人工智能·python·ai作画
AndrewHZ42 分钟前
【图像处理基石】图像匹配技术:从原理到实践,OpenCV实现与进阶方向
图像处理·人工智能·opencv·图像匹配·算法原理
数新网络1 小时前
Data Ingestion: Architectural Patterns
python·command-pattern
顾安r1 小时前
10.25 脚本 整理2024全年 GITHUB每周热门项目
linux·c语言·python·github·bash
新子y2 小时前
【小白笔记】 while 与 for + break 的比较分析
笔记·python
kida_yuan3 小时前
【从零开始】17. 中文摘要提取工具
python·算法·数据分析
chinesegf3 小时前
Docker篇2-用python运行项目和docker运行冲突问题
python·docker·容器
ZhengEnCi3 小时前
Python 特点和版本完全指南 - 从零基础到选择最适合的编程语言
python
B站计算机毕业设计之家3 小时前
计算机视觉:基于YOLOv8/YOLOv7/YOLOv6/YOLOv5的零售柜商品检测识别系统(Python+PySide6界面+训练代码)(源码+文档)✅
人工智能·深度学习·opencv·yolo·计算机视觉·零售·1024程序员节