python OpenCV 按时间间隔截取视频中任意片段帧

通过视频制作相应的数据集，截取视频中的任意片段帧。 

使用python截取视频中的任意片段帧，直接上代码： 

import cv2
import os

# 定义视频文件路径和保存图像文件路径
video_path = "video.mp4"
save_path = "data"

if not os.path.exists(save_path):
    os.makedirs(save_path)

# 定义起始和结束时间，以及帧数间隔
start_time = 10  # 单位为秒
end_time = 20  # 单位为秒
interval = 1  # 选取帧的时间间隔，单位为秒

# 打开视频文件
cap = cv2.VideoCapture(video_path)

# 从指定的时间开始读取视频帧
cap.set(cv2.CAP_PROP_POS_MSEC, start_time * 1000)

# 定义计数器变量
i = 0

# 读取视频帧，并保存每个间隔的帧为图像文件
while cap.isOpened():
    ret, frame = cap.read()
    if ret:
        # 判断当前时间是否在指定的时间段内
        current_time = cap.get(cv2.CAP_PROP_POS_MSEC) / 1000
        if current_time >= end_time:
            break
        if current_time >= start_time and (current_time - start_time) % interval == 0:
            # 保存当前帧为图像文件
            out_file_name = "video2_" + str(i).zfill(5) + ".png"
            out_file_path = os.path.join(save_path, out_file_name)
            cv2.imwrite(out_file_path, frame)
            i += 1
            print("processed frame:", i)
    else:
        break

# 关闭视频文件
cap.release()