Python-PyQt5+OpenCV获取数据流（视频流）

Posted 2021-05-21 Devil genius

在这里我们基于Python-Windows+Pycharm+PyQt5基础环境配置，1000%解决你的环境问题（一）的学习，我们搭建好PyQt5的基础环境，这里我们将进行制作一个简单的软件；

基础环境：

• Windows10
• pycharm2019.3.4
• python3.7.3
• PyQt5
• OpenCV4.5.1

这里我们将使用PyQt5创建一个获取摄像头的软件；
废话不多说，上代码

from PyQt5.QtCore import *
from PyQt5.QtWidgets import *
from PyQt5.QtGui import *
import cv2
import sys


class Example(QWidget):
    def __init__(self):
        super().__init__()
        self.initUI()
        self.initTimer()

    def initTimer(self):
        self.timer = QTimer(self)
        self.timer.timeout.connect(self.show_pic)

    def show_pic(self):
        ret, img = self.vc.read()
        if not ret:
            print('文件读取错误!\\n')
            return
        cv2.flip(img, 1, img)
        cur_frame = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
        heigt, width = cur_frame.shape[:2]
        pixmap = QImage(cur_frame, width, heigt, QImage.Format_RGB888)
        pixmap = QPixmap.fromImage(pixmap)
        self.lbl.setPixmap(pixmap)

    def openCamera(self):
        self.lbl.setEnabled(True)
        self.vc = cv2.VideoCapture(0)
        self.openCameraBtn.setEnabled(False)
        self.closeCameraBtn.setEnabled(True)
        self.timer.start(100)

    def closeCamera(self):
        self.vc.release()
        self.openCameraBtn.setEnabled(True)
        self.closeCameraBtn.setEnabled(False)
        self.QLable_close()
        self.timer.stop()

    def initUI(self):
        self.openCameraBtn = QPushButton('打开摄像头')
        self.openCameraBtn.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
        self.openCameraBtn.clicked.connect(self.openCamera)
        self.closeCameraBtn = QPushButton('关闭摄像头')
        self.closeCameraBtn.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
        self.closeCameraBtn.clicked.connect(self.closeCamera)
        self.openCameraBtn.setEnabled(True)
        self.closeCameraBtn.setEnabled(False)
        self.lbl = QLabel(self)
        self.lbl.resize(200, 100)
        self.hbox = QHBoxLayout(self)
        self.hbox.addWidget(self.lbl)

        self.vbox = QVBoxLayout(self)
        self.vbox.addWidget(self.openCameraBtn)
        self.vbox.addWidget(self.closeCameraBtn)
        self.hbox.addLayout(self.vbox)

        self.setLayout(self.hbox)
        self.QLable_close()
        self.move(300, 300)
        self.setWindowTitle('OPENCV')
        self.setGeometry(300, 300, 500, 250)
        self.show()

    def QLable_close(self):
        self.lbl.setStyleSheet("background:black;")
        self.lbl.setPixmap(QPixmap())

    def start(self):
        self.timer.start(100)


if __name__ == '__main__':
    app = QApplication(sys.argv)
    ex = Example()
    sys.exit(app.exec_())

在这里我们点击Button（打开摄像头）将会在黑色窗口中看见我们的图像，
再次点击Button（关闭摄像头），将会关闭摄像头

这个只是一个简单的实验，让我们熟悉熟悉PyQt5的基本组件！

希望这篇博文对你有用，如有问题请及时联系博主！
谢谢点赞评论！