这是上述问题的解决方案：
import sysimport scipy.io as siofrom PyQt4 import QtGui, QtCoreimport cv2class VideoCapture(QtGui.QWidget): def __init__(self, filename, parent): super(QtGui.QWidget, self).__init__() self.cap = cv2.VideoCapture(str(filename)) self.video_frame = QtGui.QLabel() parent.layout.addWidget(self.video_frame) def nextframeSlot(self): ret, frame = self.cap.read() frame = cv2.cvtColor(frame, cv2.cv.CV_BGR2RGB) img = QtGui.QImage(frame, frame.shape[1], frame.shape[0], QtGui.QImage.Format_RGB888) pix = QtGui.QPixmap.fromImage(img) self.video_frame.setPixmap(pix) def start(self): self.timer = QtCore.QTimer() self.timer.timeout.connect(self.nextframeSlot) self.timer.start(1000.0/30) def pause(self): self.timer.stop() def deleteLater(self): self.cap.release() super(QtGui.QWidget, self).deleteLater()class VideoDisplayWidget(QtGui.QWidget): def __init__(self,parent): super(VideoDisplayWidget, self).__init__(parent) self.layout = QtGui.QFormLayout(self) self.startButton = QtGui.QPushButton('Start', parent) self.startButton.clicked.connect(parent.startCapture) self.startButton.setFixedWidth(50) self.pauseButton = QtGui.QPushButton('Pause', parent) self.pauseButton.setFixedWidth(50) self.layout.addRow(self.startButton, self.pauseButton) self.setLayout(self.layout)class ControlWindow(QtGui.QMainWindow): def __init__(self): super(ControlWindow, self).__init__() self.setGeometry(50, 50, 800, 600) self.setWindowTitle("PyTrack") self.capture = None self.matPosFileName = None self.videoFileName = None self.positionData = None self.updatedPositionData = {'red_x':[], 'red_y':[], 'green_x':[], 'green_y': [], 'distance': []} self.updatedMatPosFileName = None self.isVideoFileLoaded = False self.isPositionFileLoaded = False self.quitAction = QtGui.QAction("&Exit", self) self.quitAction.setShortcut("Ctrl+Q") self.quitAction.setStatusTip('Close The App') self.quitAction.triggered.connect(self.closeApplication) self.openMatFile = QtGui.QAction("&Open Position File", self) self.openMatFile.setShortcut("Ctrl+Shift+T") self.openMatFile.setStatusTip('Open .mat File') self.openMatFile.triggered.connect(self.loadPosMatFile) self.openVideoFile = QtGui.QAction("&Open Video File", self) self.openVideoFile.setShortcut("Ctrl+Shift+V") self.openVideoFile.setStatusTip('Open .h264 File') self.openVideoFile.triggered.connect(self.loadVideoFile) self.mainMenu = self.menuBar() self.fileMenu = self.mainMenu.addMenu('&File') self.fileMenu.addAction(self.openMatFile) self.fileMenu.addAction(self.openVideoFile) self.fileMenu.addAction(self.quitAction) self.videoDisplayWidget = VideoDisplayWidget(self) self.setCentralWidget(self.videoDisplayWidget) def startCapture(self): if not self.capture and self.isPositionFileLoaded and self.isVideoFileLoaded: self.capture = VideoCapture(self.videoFileName, self.videoDisplayWidget) self.videoDisplayWidget.pauseButton.clicked.connect(self.capture.pause) self.capture.start() def endCapture(self): self.capture.deleteLater() self.capture = None def loadPosMatFile(self): try: self.matPosFileName = str(QtGui.QFileDialog.getOpenFileName(self, 'Select .mat position File')) self.positionData = sio.loadmat(self.matPosFileName) self.isPositionFileLoaded = True except: print "Please select a .mat file" def loadVideoFile(self): try: self.videoFileName = QtGui.QFileDialog.getOpenFileName(self, 'Select .h264 Video File') self.isVideoFileLoaded = True except: print "Please select a .h264 file" def closeApplication(self): choice = QtGui.QMessageBox.question(self, 'Message','Do you really want to exit?',QtGui.QMessageBox.Yes | QtGui.QMessageBox.No) if choice == QtGui.QMessageBox.Yes: print("Closing....") sys.exit() else: passif __name__ == '__main__': import sys app = QtGui.QApplication(sys.argv) window = ControlWindow() window.show() sys.exit(app.exec_())
