如何在python GUI中显示cv2视频?

时间:2019-07-17 12:41:37

标签: python opencv pyqt pyqt5 cv2

我正在使用Python / PyQt5创建一个GUI,该GUI应该在同一窗口中显示视频以及其他小部件。我已经尝试了不同的方法来解决此问题,但Stil似乎无法使其正常工作。

方法1: 使用OpenCV / cv2在像素图中添加视频只会显示视频的第一帧。

方法2: 我设法用cv2播放了视频,但是它在新窗口中打开。

方法3: 我也尝试使用QVideoWidget,但是显示空白屏幕,并且视频无法播放。

# only shows an image from the video, but in the correct place
        cap = cv2.VideoCapture('video.mov')
        ret, frame = cap.read()
        if ret:
            frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
            img = QImage(frame, frame.shape[1], frame.shape[0], QImage.Format_RGB888)
            pix = QPixmap.fromImage(img)
            pix = pix.scaled(600, 400, Qt.KeepAspectRatio, Qt.SmoothTransformation)
            self.ui.label_7.setPixmap(pix)

        # opens new window
        cap = cv2.VideoCapture('video.mov')

        while (cap.isOpened()):
            ret, frame = cap.read()

            self.ui.frame = cv2.imshow('frame', frame)
            if cv2.waitKey(1) & 0xFF == ord('q'):
                break

        cap.release()

        # shows a blank screen 
        self.mediaPlayer = QMediaPlayer(None, QMediaPlayer.VideoSurface)
        videoWidget = self.ui.vid_widget
        self.mediaPlayer.setVideoOutput(videoWidget)
        self.mediaPlayer.setMedia(QMediaContent(QUrl.fromLocalFile('video.mov')))

在如何使视频在另一个窗口小部件/在同一窗口内播放的任何帮助将不胜感激。

1 个答案:

答案 0 :(得分:0)

while循环内,您需要再次将frame转换为QPixmap,类似于上面的操作,然后更新ui

cap = cv2.VideoCapture('video.mov')

while (cap.isOpened()):
    ret, frame = cap.read()
    if not ret:
        break

    frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
    img = QImage(frame, frame.shape[1], frame.shape[0], QImage.Format_RGB888)
    pix = QPixmap.fromImage(img)
    pix = pix.scaled(600, 400, Qt.KeepAspectRatio, Qt.SmoothTransformation)
    self.ui.label_7.setPixmap(pix)    

    self.ui.frame = pix  # or img depending what `ui.frame` needs

    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()