一、引言
在Python-OpenCV中显示图像时调用的是一个单独的窗口,有时我们需要将这些图像显示在PyQt的图形化界面上,这样就可以将整个图像显示与PyQt图形化界面进行整合。但OpenCV格式的图像和PyQt格式的图像并不同,这就需要进行转换。
二、背景知识
- Python-OpenCV的图像是BGR格式的,而PyQt图像格式是RGB格式的,二者需要转换;
- 为了快速转换,图像必须基于内存进行操作;
- PyQt的QImage类可以从内存数组构建;
- OpenCV可以读取视频图像,使用waitKey可以实现休眠特定时长而不影响系统消息处理。
关于PyQt和OpenCV之间的图像转换请参考《Python-OpenCV中图像颜色空间转换》。
三、案例
下面的案例读取一个视频文件的图像进行显示,如果再叠加一个音频播放的功能,就实现了一个视频播放器。
3.1、设计图形化界面
该图形界面非常简单,包含了一个仅有“ShowImg”的菜单和对应工具栏,一个名为ImgDisp的标签对象用于显示图像(蓝色标记部分)。使用PyUIC生成的界面对象代码如下:
from PyQt5 import QtCore, QtGui, QtWidgetsclass Ui_MainWindow(object): def setupUi(self, MainWindow): MainWindow.setObjectName("MainWindow") MainWindow.resize(625, 430) self.centralwidget = QtWidgets.QWidget(MainWindow) self.centralwidget.setObjectName("centralwidget") self.ImgDisp = QtWidgets.QLabel(self.centralwidget) self.ImgDisp.setGeometry(QtCore.QRect(0, 0, 54, 12)) self.ImgDisp.setObjectName("ImgDisp") MainWindow.setCentralWidget(self.centralwidget) self.menubar = QtWidgets.QMenuBar(MainWindow) self.menubar.setGeometry(QtCore.QRect(0, 0, 625, 17)) self.menubar.setObjectName("menubar") self.menushowImg = QtWidgets.QMenu(self.menubar) self.menushowImg.setObjectName("menushowImg") MainWindow.setMenuBar(self.menubar) self.statusbar = QtWidgets.QStatusBar(MainWindow) self.statusbar.setObjectName("statusbar") MainWindow.setStatusBar(self.statusbar) self.toolBar = QtWidgets.QToolBar(MainWindow) self.toolBar.setObjectName("toolBar") MainWindow.addToolBar(QtCore.Qt.TopToolBarArea, self.toolBar) self.actionshowImg = QtWidgets.QAction(MainWindow) self.actionshowImg.setObjectName("actionshowImg") self.menushowImg.addAction(self.actionshowImg) self.menubar.addAction(self.menushowImg.menuAction()) self.toolBar.addAction(self.actionshowImg) self.retranslateUi(MainWindow) QtCore.QMetaObject.connectSlotsByName(MainWindow) def retranslateUi(self, MainWindow): _translate = QtCore.QCoreApplication.translate MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow")) self.ImgDisp.setText(_translate("MainWindow", ".")) self.menushowImg.setTitle(_translate("MainWindow", "menu")) self.toolBar.setWindowTitle(_translate("MainWindow", "toolBar")) self.actionshowImg.setText(_translate("MainWindow", "showImg"))
3.2、主程序
import cv2,sysfrom PyQt5 import QtGui,QtWidgets,QtCoreimport mainWindef cvImgtoQtImg(cvImg): #定义opencv图像转PyQt图像的函数 QtImgBuf = cv2.cvtColor(cvImg, cv2.COLOR_BGR2BGRA) QtImg = QtGui.QImage(QtImgBuf.data, QtImgBuf.shape[1], QtImgBuf.shape[0], QtGui.QImage.Format_RGB32) return QtImgclass mainwin(QtWidgets.QMainWindow,mainWin.Ui_MainWindow): def __init__(self): super().__init__() self.setupUi(self) self.bClose = False self.actionshowImg.triggered.connect(self.playVideoFile) #建立菜单点击的信号与方法playVideoFile连接 def playVideoFile(self): #播放影片 cap = cv2.VideoCapture(r'f:videomydream.mp4') #打开影片 fps = 24 if not cap.isOpened(): print("Cannot open Video File") exit() while not self.bClose: ret, frame = cap.read() #逐帧读取影片 if not ret: if frame is None: print("The video has end.") else: print("Read video error!") break QtImg = cvImgtoQtImg(frame) #将帧数据转换为PyQt图像格式 self.ImgDisp.setPixmap(QtGui.QPixmap.fromImage(QtImg)) #在ImgDisp显示图像 size = QtImg.size() self.ImgDisp.resize(size)#根据帧大小调整标签大小 self.ImgDisp.show() #刷新界面 cv2.waitKey(int(1000/fps)) #休眠一会,确保每秒播放fps帧 # 完成所有操作后,释放捕获器 cap.release()if __name__=='__main__': app = QtWidgets.QApplication(sys.argv) w = mainwin() w.show() sys.exit(app.exec_())
注意:
本文的实现方法存在不足,相关完善方案请见《OpenCV-Python图像转换为PyQt图像的变形及花屏问题研究》。
3.3、运行程序
初始界面
点击showImg,开始播放视频:
关于PyQt的使用请参考付费专栏《使用PyQt开发图形界面Python应用》,专栏文件目录《使用PyQt开发图形界面Python应用专栏目录》。
也可以参考免费专栏《PyQt入门知识》,专栏文件目录《使用PyQt进行Python图形界面程序开发文章目录》。