核心代码:
使用Labe组件显示图像,核心部分的代码如下:
ret, frame = self.video_stream.read() frame=cv2.resize(frame,(640,480),interpolation=cv2.INTER_AREA) frame=cv2.cvtColor(frame,cv2.COLOR_BGR2RGB) #cv2.imshow('test',frame) #cv2.waitKey(10) self.Qframe=QImage(frame.data,frame.shape[1],frame.shape[0],frame.shape[1]*3,QImage.Format_RGB888) #print(Qframe) #pix = QPixmap(Qframe).scaled(frame.shape[1], frame.shape[0]) #self.setPixmap(pix) #QRect qq(20,50,self.img.width,self.img.height) self.img_label.setPixmap(QPixmap.fromImage(self.Qframe)) #self.painter.drawImage(QPoint(20,50),Qframe) #print(Qframe) self.update()
程序最后的效果:
整个main.py文件代码如下:
import sys
from PyQt5.QtWidgets import QApplication,QMainWindow,QFileDialog
import MainWindow
from PyQt5.QtCore import *
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
from PyQt5 import QtWidgets, QtCore, QtGui
import cv2
class MainCode(QMainWindow, MainWindow.Ui_MainWindow):
def __init__(self):
QMainWindow.__init__(self)
MainWindow.Ui_MainWindow.__init__(self)
self.setupUi(self)
self.pushButton.clicked.connect(self.on_video)
self.open_flag=False
self.video_stream=cv2.VideoCapture('E:/dog_task_430.mp4')
self.painter = QPainter(self)
def on_video(self):
if self.open_flag:
self.pushButton.setText('open')
else:
self.pushButton.setText('close')
self.open_flag = bool(1-self.open_flag)#
def paintEvent(self, a0: QtGui.QPaintEvent):
if self.open_flag:
ret, frame = self.video_stream.read()
frame=cv2.resize(frame,(640,480),interpolation=cv2.INTER_AREA)
frame=cv2.cvtColor(frame,cv2.COLOR_BGR2RGB)
#cv2.imshow('test',frame)
#cv2.waitKey(10)
self.Qframe=QImage(frame.data,frame.shape[1],frame.shape[0],frame.shape[1]*3,QImage.Format_RGB888)
#print(Qframe)
#pix = QPixmap(Qframe).scaled(frame.shape[1], frame.shape[0])
#self.setPixmap(pix)
#QRect qq(20,50,self.img.width,self.img.height)
self.img_label.setPixmap(QPixmap.fromImage(self.Qframe))
#self.painter.drawImage(QPoint(20,50),Qframe)
#print(Qframe)
self.update()
if __name__ == '__main__':
app = QApplication(sys.argv)
md = MainCode()
md.show()
sys.exit(app.exec_())
MainWindow.ui转换为MainWindow.py代码如下:
# -*- coding: utf-8 -*-
# Form implementation generated from reading ui file 'MainWindow.ui'
#
# Created by: PyQt5 UI code generator 5.13.0
#
# 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(713, 629)
self.centralwidget = QtWidgets.QWidget(MainWindow)
self.centralwidget.setObjectName("centralwidget")
self.label = QtWidgets.QLabel(self.centralwidget)
self.label.setGeometry(QtCore.QRect(20, 20, 101, 31))
self.label.setObjectName("label")
self.pushButton = QtWidgets.QPushButton(self.centralwidget)
self.pushButton.setGeometry(QtCore.QRect(570, 540, 93, 28))
self.pushButton.setObjectName("pushButton")
self.widget = QtWidgets.QWidget(self.centralwidget)
self.widget.setGeometry(QtCore.QRect(20, 50, 640, 480))
self.widget.setObjectName("widget")
self.img_label = QtWidgets.QLabel(self.widget)
self.img_label.setGeometry(QtCore.QRect(0, 0, 640, 480))
self.img_label.setObjectName("img_label")
MainWindow.setCentralWidget(self.centralwidget)
self.menubar = QtWidgets.QMenuBar(MainWindow)
self.menubar.setGeometry(QtCore.QRect(0, 0, 713, 26))
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.label.setText(_translate("MainWindow", "videostream"))
self.pushButton.setText(_translate("MainWindow", "open"))
self.img_label.setText(_translate("MainWindow", " 视频播放区域"))