PyQt的視頻完美,可實現實時、本地、暫停、繼續、停止等功能,且無卡退、程序崩潰等情況

使用pyqt來讀取本地視頻和實時視頻

1.首先使用創建ui文件,樣式如下

組成:
lable:顯示視頻
Open:打開文件或者攝像頭並且顯示,Close爲取消
lable:顯示文件
GrouBox:爲了保證多個QRadioButton只選中一個,(注:我繼承的的是c#還要java的思想,這裏邊到底需不需要這個控件我也沒試)

在這裏插入圖片描述

先保存爲DisplayUI.ui文件然後再轉成 DisplayUI.py文件
# -*- coding: utf-8 -*-

# Form implementation generated from reading ui file 'DisplayUI.ui'
#
# Created by: PyQt5 UI code generator 5.14.2
#
# WARNING! All changes made in this file will be lost!


from PyQt5 import QtCore, QtGui, QtWidgets


class Ui_MainWindow(object):
    def setupUi(self, MainWindow):
        MainWindow.setObjectName("MainWindow")
        MainWindow.resize(800, 600)
        self.centralwidget = QtWidgets.QWidget(MainWindow)
        self.centralwidget.setObjectName("centralwidget")
        self.DispalyLabel = QtWidgets.QLabel(self.centralwidget)
        self.DispalyLabel.setGeometry(QtCore.QRect(0, 0, 681, 451))
        self.DispalyLabel.setText("")
        self.DispalyLabel.setObjectName("DispalyLabel")
        self.groupBox = QtWidgets.QGroupBox(self.centralwidget)
        self.groupBox.setGeometry(QtCore.QRect(60, 460, 120, 80))
        self.groupBox.setObjectName("groupBox")
        self.radioButtonCam = QtWidgets.QRadioButton(self.groupBox)
        self.radioButtonCam.setGeometry(QtCore.QRect(10, 20, 89, 17))
        self.radioButtonCam.setObjectName("radioButtonCam")
        self.radioButtonFile = QtWidgets.QRadioButton(self.groupBox)
        self.radioButtonFile.setGeometry(QtCore.QRect(10, 40, 89, 17))
        self.radioButtonFile.setCheckable(True)
        self.radioButtonFile.setChecked(True)
        self.radioButtonFile.setObjectName("radioButtonFile")
        self.Open = QtWidgets.QPushButton(self.centralwidget)
        self.Open.setGeometry(QtCore.QRect(220, 490, 75, 21))
        self.Open.setObjectName("Open")
        self.Close = QtWidgets.QPushButton(self.centralwidget)
        self.Close.setGeometry(QtCore.QRect(380, 490, 75, 23))
        self.Close.setObjectName("Close")

        self.First= QtWidgets.QPushButton(self.centralwidget)
        self.First.setGeometry(QtCore.QRect(500, 490, 75, 23))
        self.First.setObjectName("First")
        MainWindow.setCentralWidget(self.centralwidget)
        self.menubar = QtWidgets.QMenuBar(MainWindow)
        self.menubar.setGeometry(QtCore.QRect(0, 0, 800, 22))
        self.menubar.setObjectName("menubar")
        MainWindow.setMenuBar(self.menubar)
        self.statusbar = QtWidgets.QStatusBar(MainWindow)
        self.statusbar.setObjectName("statusbar")
        MainWindow.setStatusBar(self.statusbar)

        self.retranslateUi(MainWindow)
        QtCore.QMetaObject.connectSlotsByName(MainWindow)

    def retranslateUi(self, MainWindow):
        _translate = QtCore.QCoreApplication.translate
        MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow"))
        self.groupBox.setTitle(_translate("MainWindow", "GroupBox"))
        self.radioButtonCam.setText(_translate("MainWindow", "攝像模式"))
        self.radioButtonFile.setText(_translate("MainWindow", "本地文件模式"))
        self.Open.setText(_translate("MainWindow", "Open"))
        self.Close.setText(_translate("MainWindow", "Close"))
        self.First.setText(_translate("MainWindow", "暫停||繼續"))


2.功能事件

寫一個VideoDisplay.py文件,用來生成各個控件的事件和槽,,以及創建線程,
註明:內部代碼邏輯已經改好,,實現了暫停和繼續的功能,還有就是解決了視頻播放完畢程序各個按鈕事件失靈的問題,以及再開啓攝像頭的模式下再點擊本地文件模式會崩潰的問題。。好了,大概就這麼多。下面上螢火(硬貨)。

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2020/4/27 19:40
# @Author  : 沐白
# @Site    :
# @File    : VideoDisplay.py
# @Software: PyCharm
import cv2
import threading
from PyQt5.QtCore import QFile
from PyQt5.QtWidgets import QFileDialog, QMessageBox
from PyQt5.QtGui import QImage, QPixmap
import time
import os

class Display:
    def __init__(self, ui, mainWnd):
        self.ui = ui
        self.mainWnd = mainWnd
        # 默認視頻源爲相機
        self.ui.radioButtonCam.setChecked(True)
        self.isCamera = True
        self.a = 0
        self.b=0
        # 信號槽設置
        ui.Open.clicked.connect(self.Open)
        ui.Close.clicked.connect(self.Close)
        ui.radioButtonCam.clicked.connect(self.radioButtonCam)
        ui.radioButtonFile.clicked.connect(self.radioButtonFile)
        ui.First.clicked.connect(self.suspend_continue)

        # 創建一個關閉事件並設爲未觸發
        self.continueEvent1 = threading.Event()
        self.continueEvent1.clear()

        self.stopEvent = threading.Event()
        self.stopEvent.clear()

    def radioButtonCam(self):
        if self.a==0:
            self.isCamera = True

    def suspend_continue(self):
        self.continueEvent1.set()
    def radioButtonFile(self):
        if self.a==0:
            self.isCamera = False

    # def pause(self):
    #     self.__flag.clear()  # 設置爲False, 讓線程阻塞
    #     print("pause")
    #
    # def resume(self):
    #     self.__flag.set()  # 設置爲True, 讓線程停止阻塞
    #     print("resume")

    def Open(self):
        self.fileName=""
        if not self.isCamera:
            self.fileName, self.fileType = QFileDialog.getOpenFileName(self.mainWnd, 'Choose file', '',"MP4Files(*.mp4);;AVI Files(*.avi)" )
            self.cap = cv2.VideoCapture(self.fileName)
            self.frameRate = self.cap.get(cv2.CAP_PROP_FPS)
        else:
            # 下面兩種rtsp格式都是支持的
            #  cap = cv2.VideoCapture("rtsp://admin:[email protected]/main/Channels/1")
             self.a=1
             self.cap = cv2.VideoCapture(0)

        # 創建視頻顯示線程
        if  (self.fileName != "") or(self.a==1) :
            th = threading.Thread(target=self.Display)
            th.start()


    def Close(self):
        # 關閉事件設爲觸發,關閉視頻播放
        self.a = 0
        self.stopEvent.set()


    def Display(self):
        self.ui.Open.setEnabled(False)
        self.ui.Close.setEnabled(True)
        self.ui.First.setEnabled(True)

        while self.cap.isOpened() and True:
            success, frame = self.cap.read()
            # RGB轉BGR
            print(success)
            if success==False:
                print("play finished")  # 判斷本地文件播放完畢
                break
            frame = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR)

            img = QImage(frame.data, frame.shape[1], frame.shape[0], QImage.Format_RGB888)
            self.ui.DispalyLabel.setPixmap(QPixmap.fromImage(img))

            if self.isCamera:
                cv2.waitKey(1)
            else:
                cv2.waitKey(int(1000 / self.frameRate))
            # 判斷關閉事件是否已觸發

            if True==self.continueEvent1.is_set():
                self.continueEvent1.clear()
                self.b=1
                while self.b==1:
                    if True == self.continueEvent1.is_set():
                        self.continueEvent1.clear()
                        self.b=0
            if True == self.stopEvent.is_set():
                # 關閉事件置爲未觸發,清空顯示label
                break
        self.cap.release()
        self.stopEvent.clear()
        self.ui.DispalyLabel.clear()
        self.ui.Close.setEnabled(False)
        self.ui.Open.setEnabled(True)
        self.ui.DispalyLabel.setText("1212")

3.生成主函數

至於有人問爲什麼不把這三個文件寫在一起???
答案是:能寫在一起,但是你想想一個文件劃分爲每個功能不更好用嗎,最重要的不會吧自己繞迷糊,

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2020/4/26 19:36
# @Author  : 沐白
# @Site    : 
# @File    : Main.py
# @Software: PyCharm


# import sys
# from PyQt5.QtWidgets import QApplication, QMainWindow
# import asa
#
#
#
#
# if __name__ == '__main__':
#     app = QApplication(sys.argv)
#     MainWindow = QMainWindow()
#     ui = asa.Ui_MainWindow()
#     ui.setupUi(MainWindow)
#     MainWindow.show()
#     sys.exit(app.exec_())


import sys
import DisplayUI
from PyQt5.QtWidgets import QApplication, QMainWindow
from VideoDisplay import Display

if __name__ == '__main__':
    app = QApplication(sys.argv)
    mainWnd = QMainWindow()
    ui = DisplayUI.Ui_MainWindow()
    # 可以理解成將創建的 ui 綁定到新建的 mainWnd 上
    ui.setupUi(mainWnd)
    display = Display(ui, mainWnd)
    mainWnd.show()

    sys.exit(app.exec_())

最後呢希望大家收藏走起一波,後續會有pyqt,tensorflow,爬蟲,python基礎的各個模塊,,
更重要的是我都改bug改到0:34了,,大家給按排一波》》》

在這裏插入圖片描述

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章