pyqt5 pyqt5+opencv 实现读取视频数据_pyqt opencv 视频

        else:
            break
    capture.release()
    cv2.destroyAllWindows()
else:
    print("摄像头或视频读取失败")

## 2、openCV集成pyqt5读取视频数据

import cv2
import numpy as np
import sys
from PyQt5.QtGui import *
from PyQt5.QtCore import *
from PyQt5.QtWidgets import *

class Video():
def __init__(self, capture):
self.capture = capture
self.currentFrame = np.array([])

def captureNextFrame(self):
    ret, readFrame = self.capture.read()
    if (ret == True):
        self.currentFrame = cv2.resize(readFrame, (960, 540))


def convertFrame(self):
    try:

        height, width, channel = self.currentFrame.shape
        bytesPerLine = 3 \* width
        qImg = QImage(self.currentFrame.data, width, height, bytesPerLine,
                           QImage.Format_RGB888).rgbSwapped()
        qImg = QPixmap.fromImage(qImg)
        return qImg
    except:
        return None

class win(QMainWindow):
def __init__(self, parent=None):
super().__init__()
self.setGeometry(250, 80, 800, 600) # 从屏幕(250，80)开始建立一个800*600的界面
self.setWindowTitle(‘camera’)
self.videoPath = “./dataSet/3700000000003_13-38-20.055.mp4”
self.video = Video(cv2.VideoCapture(self.videoPath))
self._timer = QTimer(self)
self._timer.timeout.connect(self.play)
self._timer.start(27)
self.update()
self.videoFrame = QLabel(‘VideoCapture’)
self.videoFrame.setAlignment(Qt.AlignCenter)
self.setCentralWidget(self.videoFrame) # 设置图像数据填充控件

def play(self):
    try:
        self.video.captureNextFrame()
        self.videoFrame.setPixmap(self.video.convertFrame())
        self.videoFrame.setScaledContents(True)     # 设置图像自动填充控件
    except TypeError:
        print('No Frame')

if name == ‘__main__’:
app = QApplication(sys.argv)
win = win()
win.show()
sys.exit(app.exec_())

![在这里插入图片描述](https://img-blog.csdnimg.cn/7aee6ddb46be44c5b1e626bea0eb8121.png?x-oss-process=image/watermark,type_d3F5LXplbmhlaQ,shadow_50,text_Q1NETiBA6YOt5bqG5rGd,size_20,color_FFFFFF,t_70,g_se,x_16)


### 界面美化版：

import sys
import os
import cv2

from PyQt5 import QtCore, QtGui, QtWidgets
from PyQt5.QtWidgets import *
from PyQt5.QtCore import *
from PyQt5.QtGui import QPalette, QBrush, QPixmap

class Ui_MainWindow(QtWidgets.QWidget):
def __init__(self, parent=None):
super(Ui_MainWindow, self).__init__(parent)

    self.timer_camera = QtCore.QTimer()  # 初始化定时器
    self.cap = cv2.VideoCapture()  # 初始化摄像头
    self.CAM_NUM = r"D:\PycharmProjects\ele\_good\_pyqt5\dataSet\00.flv"
    self.set\_ui()
    self.slot\_init()
    self.__flag_work = 0
    self.x = 0
    self.count = 0

def set\_ui(self):
    self.__layout_main = QtWidgets.QHBoxLayout()  # 采用QHBoxLayout类，按照从左到右的顺序来添加控件
    self.__layout_fun_button = QtWidgets.QHBoxLayout()
    self.__layout_data_show = QtWidgets.QVBoxLayout()  # QVBoxLayout类垂直地摆放小部件

    self.button_open_camera = QtWidgets.QPushButton(u'打开相机')
    self.button_close = QtWidgets.QPushButton(u'退出')

    # button颜色修改
    button_color = [self.button_open_camera, self.button_close]
    for i in range(2):
        button_color[i].setStyleSheet("QPushButton{color:black}"
                                       "QPushButton:hover{color:red}"
                                       "QPushButton{background-color:rgb(78,255,255)}"
                                       "QpushButton{border:2px}"
                                       "QPushButton{border\_radius:10px}"
                                       "QPushButton{padding:2px 4px}")

    self.button_open_camera.setMinimumHeight(50)
    self.button_close.setMinimumHeight(50)

    # move()方法是移动窗口在屏幕上的位置到x = 500，y = 500的位置上
    self.move(500, 500)

    # 信息显示
    self.label_show_camera = QtWidgets.QLabel()
    self.label_move = QtWidgets.QLabel()
    self.label_move.setFixedSize(100, 100)

    self.label_show_camera.setFixedSize(641, 481)
    self.label_show_camera.setAutoFillBackground(False)

    self.__layout_fun_button.addWidget(self.button_open_camera)
    self.__layout_fun_button.addWidget(self.button_close)
    self.__layout_fun_button.addWidget(self.label_move)

    self.__layout_main.addLayout(self.__layout_fun_button)
    self.__layout_main.addWidget(self.label_show_camera)

    self.setLayout(self.__layout_main)
    self.label_move.raise\_()            # 设置控件在最上层
    self.setWindowTitle(u'摄像头')

    '''
    # 设置背景颜色
    palette1 = QPalette()
    palette1.setBrush(self.backgroundRole(),QBrush(QPixmap('background.jpg')))
    self.setPalette(palette1)
    '''

def slot\_init(self):  # 建立通信连接
    self.button_open_camera.clicked.connect(self.button_open_camera_click)
    self.timer_camera.timeout.connect(self.show_camera)
    self.button_close.clicked.connect(self.close)

def button\_open\_camera\_click(self):
    if self.timer_camera.isActive() == False:
        flag = self.cap.open(self.CAM_NUM)      # 打开摄像头操作
        if flag == False:
            msg = QtWidgets.QMessageBox.Warning(self, u'Warning', u'请检测相机与电脑是否连接正确',
                                                buttons=QtWidgets.QMessageBox.Ok,
                                                defaultButton=QtWidgets.QMessageBox.Ok)
            # if msg==QtGui.QMessageBox.Cancel:
            # pass
        else:
            self.timer_camera.start(30)
            self.button_open_camera.setText(u'关闭相机')        # 将控件内容设置为关闭
    else:
        self.timer_camera.stop()
        self.cap.release()
        self.label_show_camera.clear()
        self.button_open_camera.setText(u'打开相机')

def show\_camera(self):
    flag, self.image = self.cap.read()      # 读取摄像头数据
    show = cv2.resize(self.image, (640, 480))
    show = cv2.cvtColor(show, cv2.COLOR_BGR2RGB)
    showImage = QtGui.QImage(show.data, show.shape[1], show.shape[0], QtGui.QImage.Format_RGB888)
    self.label_show_camera.setPixmap(QtGui.QPixmap.fromImage(showImage))

def closeEvent(self, event):
    print("关闭")
    ok = QtWidgets.QPushButton()
    cancel = QtWidgets.QPushButton()
    msg = QtWidgets.QMessageBox(QtWidgets.QMessageBox.Warning, u'关闭', u'是否关闭！')
    msg.addButton(ok, QtWidgets.QMessageBox.ActionRole)
    msg.addButton(cancel, QtWidgets.QMessageBox.RejectRole)
    ok.setText(u'确定')
    cancel.setText(u'取消')
    if msg.exec\_() == QtWidgets.QMessageBox.RejectRole:
        event.ignore()
    else:
        if self.cap.isOpened():
            self.cap.release()
        if self.timer_camera.isActive():
            self.timer_camera.stop()
        event.accept()

if name == ‘__main__’:
App = QApplication(sys.argv)
win = Ui_MainWindow()
win.show()
sys.exit(App.exec_())