使用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了,,大家給按排一波》》》