pyqt5+opencv?實(shí)現(xiàn)讀取視頻數(shù)據(jù)的方法

更新時(shí)間：2022年01月25日 14:19:47 作者：郭慶汝

這篇文章主要介紹了pyqt5+opencv?實(shí)現(xiàn)讀取視頻數(shù)據(jù)的方法，本文通過實(shí)例代碼給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下

1、openCV讀取視頻數(shù)據(jù)

import cv2

if __name__ == '__main__':
    videoPath = "./dataSet/3700000000003_13-38-20.055.mp4"
    nameWindow = "Detection window"  # 窗體名稱
    cv2.namedWindow(nameWindow)  # 設(shè)置窗體
    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讀取視頻數(shù)據(jù)

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)開始建立一個(gè)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)          # 設(shè)置圖像數(shù)據(jù)填充控件
    def play(self):
            self.video.captureNextFrame()
            self.videoFrame.setPixmap(self.video.convertFrame())
            self.videoFrame.setScaledContents(True)     # 設(shè)置圖像自動(dòng)填充控件
        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()  # 初始化定時(shí)器
        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類垂直地?cái)[放小部件
        self.button_open_camera = QtWidgets.QPushButton(u'打開相機(jī)')
        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()方法是移動(dòng)窗口在屏幕上的位置到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_()            # 設(shè)置控件在最上層
        self.setWindowTitle(u'攝像頭')
        '''
        # 設(shè)置背景顏色
        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'請檢測相機(jī)與電腦是否連接正確',
                                                    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'關(guān)閉相機(jī)')        # 將控件內(nèi)容設(shè)置為關(guān)閉
        else:
            self.timer_camera.stop()
            self.cap.release()
            self.label_show_camera.clear()
            self.button_open_camera.setText(u'打開相機(jī)')
    def show_camera(self):
        flag, self.image = self.cap.read()      # 讀取攝像頭數(shù)據(jù)
        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("關(guān)閉")
        ok = QtWidgets.QPushButton()
        cancel = QtWidgets.QPushButton()
        msg = QtWidgets.QMessageBox(QtWidgets.QMessageBox.Warning, u'關(guān)閉', u'是否關(guān)閉！')
        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_())

顯示效果如下所示：

到此這篇關(guān)于pyqt5+opencv 實(shí)現(xiàn)讀取視頻數(shù)據(jù)的文章就介紹到這了,更多相關(guān)pyqt5 opencv讀取視頻數(shù)據(jù)內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: