『Python』获取视频中的关键帧

本文将利用python的opencv库来获取任意视频中的关键帧

一、安装numpy以及opencv

在终端中输入

pip install numpy
pip install opencv-python

二、上代码

新建一个文件夹，在文件夹中创建“main.py”，输入代码

import cv2

capture = cv2.VideoCapture("xxx.mp4") #此处为要提帧的视频名称
if capture.isOpened() is False:
    print("Error opening the video file!")

frame_index=0
while capture.isOpened():
    ret, frame = capture.read()
    if ret is True:
        cv2.imshow('xxx.mp4', frame)
        if cv2.waitKey(20) & 0xFF == ord('c'):
            frame_name = "video_frame_{}.png".format(frame_index)
            # 保存当前帧
            cv2.imwrite(frame_name, frame)
            frame_index += 1
        if cv2.waitKey(20) & 0xFF == ord('q'):
            break
    else:
        break
capture.release()
cv2.destroyAllWindows()

将你想要提帧的视频放入文件夹中，再将代码第二行中的“xxx.mp4"改为放入的视频名称，运行，视频就会播放，在你想要截取时按c键就可以获取当前的视频图片，按q键可以退出视频。截获的图片会保存在文件夹中。