pyqt5+opencv 实现读取视频数据的方法

这篇文章主要介绍了pyqt5+opencv 实现读取视频数据的方法,本文通过实例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下

1、openCV读取视频数据

import cv2 if __name__ == '__main__': videoPath = "./dataSet/3700000000003_13-38-20.055.mp4" nameWindow = "Detection window"  # 窗体名称 cv2.namedWindow(nameWindow)  # 设置窗体 capture = cv2.VideoCapture(videoPath) if capture.isOpened(): size = (capture.get(cv2.CAP_PROP_FRAME_WIDTH), capture.get(cv2.CAP_PROP_FRAME_HEIGHT))  # 读取帧的宽、高 speed = capture.get(cv2.CAP_PROP_FPS)  # 获得帧速 while True: ret, frame = capture.read() if ret: frame = cv2.resize(frame, (960, 540)) cv2.imshow(nameWindow, frame) if cv2.waitKey(1) & 0xFF == 27: break 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): 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_())

界面美化版:

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-600'))) 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() 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_())

显示效果如下所示:

到此这篇关于pyqt5+opencv 实现读取视频数据的文章就介绍到这了,更多相关pyqt5 opencv读取视频数据内容请搜索0133技术站以前的文章或继续浏览下面的相关文章希望大家以后多多支持0133技术站!

以上就是pyqt5+opencv 实现读取视频数据的方法的详细内容,更多请关注0133技术站其它相关文章!

赞(0) 打赏
未经允许不得转载:0133技术站首页 » python