Python-PyQt5+OpenCV获取数据流(视频流)(二)

在这里我们基于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_())

Python-PyQt5+OpenCV获取数据流(视频流)(二)_第1张图片
在这里我们点击Button(打开摄像头)将会在黑色窗口中看见我们的图像,
再次点击Button(关闭摄像头),将会关闭摄像头

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

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

你可能感兴趣的:(PyQt5,opencv,pyqt,python,pyqt5,人工智能)