记录: PYQT5+OPENCV实现RTSP/RTMP视频流播放

原文：记录: pyqt5+opencv实现rtsp/rtmp视频流播放 - 灰信网（软件开发博客聚合）

最近有一个需求，要求写一个客户端实时展示多个摄像头的视频流。
界面大致设计如下：

由于我是搞Java出生的，很想用java写客户端，但是java写的窗口程序，实在是太。。。
所以，最后还是选择python+qt来完成这个任务。

使用python 3.6
要调用摄像头，首先想到使用opencv的VideoCapture组件了（还有一种方式是使用vlc，暂时不研究了）
公司摄像头是rtsp协议的流，网络资源是rtmp协议的流（这里就用网络资源）

我的大致实现思路，比较简单：
1.窗口设计6个label
2.每个摄像头定义一个线程，需要启动6个子线程
3.每个线程里获取视频流每一帧，用label展示

第一步：
使用QtDesigner设计UI：

在PyCharm中使用qt的pyuic插件生成UI类：

第二步
编写主类：

# 作者：gc
# 日期：2020-04-09 14:06
# 描述：测试主类

import cv2, sys, os, threading, time
from PyQt5 import QtCore, QtGui, QtWidgets
from PyQt5.QtGui import QImage, QPixmap
from form.TestCamera import Ui_Form


class TestCamera(QtWidgets.QWidget, Ui_Form):
    def __init__(self):
        super(TestCamera, self).__init__()
        self.setupUi(self)
        self.init_camera()

    def init_camera(self):
        """初始化摄像头监控"""
        print("进程pid:", os.getpid())
        threading.Thread(target=Camera('rtmp://58.200.131.2:1935/livetv/hunantv', self.Camera1).display).start()
        threading.Thread(target=Camera('rtmp://58.200.131.2:1935/livetv/gxtv', self.Camera2).display).start()
        threading.Thread(target=Camera('rtmp://58.200.131.2:1935/livetv/gdtv', self.Camera3).display).start()
        threading.Thread(target=Camera('rtmp://58.200.131.2:1935/livetv/dftv', self.Camera4).display).start()
        threading.Thread(target=Camera('rtmp://58.200.131.2:1935/livetv/gxtv', self.Camera5).display).start()
        threading.Thread(target=Camera('rtmp://58.200.131.2:1935/livetv/hunantv', self.Camera6).display).start()


class Camera:
    """摄像头对象"""

    def __init__(self, url, out_label):
        """初始化方法"""
        self.url = url
        self.outLabel = out_label

    def display(self):
        """显示"""
        cap = cv2.VideoCapture(self.url)
        start_time = time.time()
        while cap.isOpened():
            success, frame = cap.read()
            if success:
                if (time.time() - start_time) > 0.1:
                    frame = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR)
                    img = QImage(frame.data, frame.shape[1], frame.shape[0], QImage.Format_RGB888)
                    self.outLabel.setPixmap(QPixmap.fromImage(img))
                    cv2.waitKey(1)
                    start_time = time.time()


if __name__ == '__main__':
    app = QtWidgets.QApplication(sys.argv)
    mainWindow = TestCamera()
    mainWindow.show()
    sys.exit(app.exec())

说明一下为什么在display里面加入(time.time() - start_time) > 0.1条件：
最开始我是没有加这些条件的，但是程序会在运行时突然挂掉，原因是因为

frame = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR)
img = QImage(frame.data, frame.shape[1], frame.shape[0], QImage.Format_RGB888)
self.outLabel.setPixmap(QPixmap.fromImage(img))

这部分代码很费cpu的资源，我电脑性能一般，4核16线程的，同时处理6个摄像头的视频流，有点吃不消（不知道是不是这个原因，但是我加了延迟，确实是减少了我cpu的开销），所以我加了一个延迟，0.1s。
说说我个人的理解吧，做这个客户端看使用场景，如果是单摄像头实时显示，那至少我这台电脑是能够保证流处理的速度，几乎不延迟和卡顿，如果是多摄像头显示，这个客户端是放在定制终端上使用，建议还是使用带GPU的终端好点，图片处理速度更快，毕竟是GPU,如果是要求随处安装的客户端，我不知道该怎么弄才能解决CPU开销大的问题。
第三步
看看实现效果吧：

我把UI类也贴出来吧：

# -*- coding: utf-8 -*-

# Form implementation generated from reading ui file 'TestCamera.ui'
#
# Created by: PyQt5 UI code generator 5.13.0
#
# WARNING! All changes made in this file will be lost!


from PyQt5 import QtCore, QtGui, QtWidgets


class Ui_Form(object):
    def setupUi(self, Form):
        Form.setObjectName("Form")
        Form.resize(797, 336)
        self.layoutWidget = QtWidgets.QWidget(Form)
        self.layoutWidget.setGeometry(QtCore.QRect(30, 20, 741, 292))
        self.layoutWidget.setObjectName("layoutWidget")
        self.gridLayout = QtWidgets.QGridLayout(self.layoutWidget)
        self.gridLayout.setContentsMargins(0, 0, 0, 0)
        self.gridLayout.setObjectName("gridLayout")
        self.Camera1 = QtWidgets.QLabel(self.layoutWidget)
        self.Camera1.setMinimumSize(QtCore.QSize(241, 141))
        self.Camera1.setMaximumSize(QtCore.QSize(241, 141))
        self.Camera1.setAutoFillBackground(True)
        self.Camera1.setText("")
        self.Camera1.setScaledContents(True)
        self.Camera1.setObjectName("Camera1")
        self.gridLayout.addWidget(self.Camera1, 0, 0, 1, 1)
        self.Camera2 = QtWidgets.QLabel(self.layoutWidget)
        self.Camera2.setMinimumSize(QtCore.QSize(241, 141))
        self.Camera2.setMaximumSize(QtCore.QSize(241, 141))
        self.Camera2.setAutoFillBackground(True)
        self.Camera2.setText("")
        self.Camera2.setScaledContents(True)
        self.Camera2.setObjectName("Camera2")
        self.gridLayout.addWidget(self.Camera2, 0, 1, 1, 1)
        self.Camera3 = QtWidgets.QLabel(self.layoutWidget)
        self.Camera3.setMinimumSize(QtCore.QSize(241, 141))
        self.Camera3.setMaximumSize(QtCore.QSize(241, 141))
        self.Camera3.setAutoFillBackground(True)
        self.Camera3.setText("")
        self.Camera3.setScaledContents(True)
        self.Camera3.setObjectName("Camera3")
        self.gridLayout.addWidget(self.Camera3, 0, 2, 1, 1)
        self.Camera4 = QtWidgets.QLabel(self.layoutWidget)
        self.Camera4.setMinimumSize(QtCore.QSize(241, 141))
        self.Camera4.setMaximumSize(QtCore.QSize(241, 141))
        self.Camera4.setAutoFillBackground(True)
        self.Camera4.setText("")
        self.Camera4.setScaledContents(True)
        self.Camera4.setObjectName("Camera4")
        self.gridLayout.addWidget(self.Camera4, 1, 0, 1, 1)
        self.Camera5 = QtWidgets.QLabel(self.layoutWidget)
        self.Camera5.setMinimumSize(QtCore.QSize(241, 141))
        self.Camera5.setMaximumSize(QtCore.QSize(241, 141))
        self.Camera5.setAutoFillBackground(True)
        self.Camera5.setText("")
        self.Camera5.setScaledContents(True)
        self.Camera5.setObjectName("Camera5")
        self.gridLayout.addWidget(self.Camera5, 1, 1, 1, 1)
        self.Camera6 = QtWidgets.QLabel(self.layoutWidget)
        self.Camera6.setMinimumSize(QtCore.QSize(241, 141))
        self.Camera6.setMaximumSize(QtCore.QSize(241, 141))
        self.Camera6.setAutoFillBackground(True)
        self.Camera6.setText("")
        self.Camera6.setScaledContents(True)
        self.Camera6.setObjectName("Camera6")
        self.gridLayout.addWidget(self.Camera6, 1, 2, 1, 1)

        self.retranslateUi(Form)
        QtCore.QMetaObject.connectSlotsByName(Form)

    def retranslateUi(self, Form):
        _translate = QtCore.QCoreApplication.translate
        Form.setWindowTitle(_translate("Form", "Form"))