如何使用 PyQt 在视频时间线上显示交互式缩略图？

python

2024-03-10 16:20:31

如何使用 PyQt 在视频时间线上显示缩略图

引言

想象一下，你正在使用视频编辑软件，你需要在不实际播放视频的情况下快速预览视频中的特定部分。这时，时间线缩略图就派上用场了。它们可以让你一目了然地查看整个视频，并快速找到想要查找的片段。

在这篇文章中，我们将逐步指导你如何使用 PyQt 和 OpenCV，在视频时间线上实现交互式缩略图功能。

步骤 1：添加 OpenCV

PyQt 本身不具备从视频中提取帧的功能，因此我们需要借助一个外部库。推荐使用 OpenCV，它是一个功能强大的计算机视觉库，可以帮助我们提取视频帧。

步骤 2：创建一个视频播放器类

为了在视频播放器中添加缩略图功能，我们需要创建一个自定义的视频播放器类。这个类将继承 PyQt 的 QMediaPlayer，并添加一些新的方法和属性来处理缩略图。

步骤 3：提取视频帧

为了创建缩略图，我们需要从视频中提取帧。我们可以使用 OpenCV 的 cv2.grabFrame() 函数来完成此操作。

步骤 4：显示缩略图

提取帧后，我们需要将其显示在时间线上。我们将创建一个 QGraphicsView 和 QGraphicsScene，将帧作为 QPixmap 添加到场景中。

步骤 5：链接时间线和缩略图

最后，我们需要将时间线和缩略图链接起来。当用户移动时间线滑块时，我们将更新缩略图以显示该位置的帧。

代码示例

以下是一个示例代码段，展示了如何实现上述步骤：

import cv2
from PyQt6.QtMultimedia import QMediaPlayer, QMediaContent
from PyQt6.QtCore import QTimer
from PyQt6.QtWidgets import QGraphicsView, QGraphicsScene, QApplication, QMainWindow, QWidget
from PyQt6.QtGui import QPixmap, QImage

class VideoPlayer(QMediaPlayer):
    def __init__(self, parent=None):
        super(VideoPlayer, self).__init__(parent)

        # 创建用于显示帧的 QGraphicsView
        self.frame_view = QGraphicsView()
        self.frame_view.setStyleSheet("border: 1px solid black")

        # 创建一个 QGraphicsScene 来存储视频帧
        self.frame_scene = QGraphicsScene()
        self.frame_view.setScene(self.frame_scene)

        # 创建一个计时器用于定期从视频中提取帧
        self.frame_timer = QTimer()
        self.frame_timer.setInterval(100)  # 每 100 毫秒提取一帧
        self.frame_timer.timeout.connect(self.update_frame)

        # 保存当前视频帧的图像
        self.current_frame = None

    def update_frame(self):
        # 获取视频的当前位置
        current_position = self.position()

        # 从视频中提取当前帧
        frame = self.get_frame(current_position)

        # 将帧转换为 QImage 以便在 QGraphicsView 中显示
        frame_image = QImage(frame, frame.shape[1], frame.shape[0], QImage.Format_RGB888)
        frame_pixmap = QPixmap.fromImage(frame_image)

        # 在 QGraphicsScene 中显示帧
        self.frame_scene.clear()
        self.frame_scene.addPixmap(frame_pixmap)

    def get_frame(self, position):
        # 将视频帧转换为毫秒
        frame_msec = position * 1000

        # 提取帧
        self.set_position(frame_msec)
        frame = self.grab_frame()

        # 将帧转换为 RGB 格式
        if frame is not None:
            frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)

        return frame

if __name__ == "__main__":
    app = QApplication(sys.argv)
    player = VideoPlayer()
    player.setMedia(QMediaContent(QUrl.fromLocalFile("my_video.mp4")))
    player.play()
    player.show()
    sys.exit(app.exec())