python视频流截图（按帧数）

一、安装opencv计算机视觉库

pip install opencv-python

二、视频流截图

1、读取视频文件，获取视频帧数

import cv2
# 视频位置
video_path = path_file_name
# 读取视频
cap = cv2.VideoCapture(video_path)
# 获取视频总帧数
frame_count = cap.get(cv2.CAP_PROP_FRAME_COUNT)

2、循环遍历帧率，并保存截图

while cap.isOpened():
    cap.set(cv2.CAP_PROP_POS_MSEC, flag)
    cap.set(cv2.CAP_PROP_POS_FRAMES, start_frame)
    ret, img = cap.read()
    cv2.waitKey(2000)
    cv2.imwrite((image_path + "image2/{}.jpg").format(flag), img)
    flag += 1
    start_frame += frame_interval
    if start_frame >= frame_count:
        break

3、全部代码

import cv2


def main(path_file_name):
    # 视频位置
    video_path = path_file_name
    # 读取视频
    cap = cv2.VideoCapture(video_path)
    # 获取视频总帧数
    frame_count = cap.get(cv2.CAP_PROP_FRAME_COUNT)
    # 截图时帧的间距，这里10是要等间距截取10张图 # 帧数
    # frame_interval = int(frame_count // 10)
    frame_interval = 90
    # 起始截取帧位置
    start_frame = 1
    # 截图存储位置
    image_path = ""
    flag = 0
    while cap.isOpened():
        cap.set(cv2.CAP_PROP_POS_MSEC, flag)
        cap.set(cv2.CAP_PROP_POS_FRAMES, start_frame)
        ret, img = cap.read()
        cv2.waitKey(2000)
        cv2.imwrite((image_path + "image2/{}.jpg").format(flag), img)
        flag += 1
        start_frame += frame_interval
        if start_frame >= frame_count:
            break
    cap.release()
    cv2.destroyAllWindows()


if __name__ == '__main__':
    # 视频地址
    file_name = "视频链接"
    main(file_name)