每隔10秒钟取连续五帧

每隔10秒钟取五帧

import cv2
import os
def extract_frames(input_file, input_file_1,input_file_2, output_dir, output_dir_1, output_dir_2, interval, num_intervals, frames_per_interval):
if not os.path.exists(output_dir):
os.makedirs(output_dir, exist_ok=True)
if not os.path.exists(output_dir_1):
os.makedirs(output_dir_1, exist_ok=True)
if not os.path.exists(output_dir_2):
os.makedirs(output_dir_2, exist_ok=True)

打开视频文件

vidcap = cv2.VideoCapture(input_file)
vidcap_1 = cv2.VideoCapture(input_file_1)
vidcap_2 = cv2.VideoCapture(input_file_2)
# 获取视频的帧率
fps = vidcap.get(cv2.CAP_PROP_FPS)
# 计算每个间隔需要跳过的帧数
skip_frames = int(fps * interval)
# 初始化计数器
count = 0
while count < num_intervals:
    print(count+1)
    for i in range(frames_per_interval):
        # 读取并保存当前帧
        success, image = vidcap.read()
        success_1, image_1 = vidcap_1.read()
        success_2, image_2 = vidcap_2.read()
        if not success or not success_1 or not success_2:
            break
        # 生成输出文件名
        # output_file = os.path.join(interval_dir, f"frame_{i+1}.jpg")
        output_file = os.path.join(output_dir, f"frame_{count+1}_{i+1}.jpg")
        output_file_1 = os.path.join(output_dir_1, f"frame_{count+1}_{i+1}.jpg")
        output_file_2 = os.path.join(output_dir_2, f"frame_{count+1}_{i+1}.jpg")
        # 保存帧为图像文件
        cv2.imwrite(output_file, image)
        cv2.imwrite(output_file_1, image_1)
        cv2.imwrite(output_file_2, image_2)
    # 跳过需要跳过的帧数
    for _ in range(skip_frames):
        vidcap.read()
        vidcap_1.read()
        vidcap_2.read()
    count += 1
# 释放资源
vidcap.release()
vidcap_1.release()
vidcap_2.release()

设置输入文件路径

input_file = “/disk2/sunyingli/1950他们正年轻/志愿军老兵讲述历史360.mp4”
input_file_1 = “/disk2/sunyingli/1950他们正年轻/志愿军老兵讲述历史1080.mp4”
input_file_2 = “/disk2/sunyingli/1950他们正年轻/志愿军老兵讲述历史4K.mp4”

设置输出目录

output_dir = “/disk2/sunyingli/1950他们正年轻/志愿军老兵讲述历史360_10s_5_test”
output_dir_1 = “/disk2/sunyingli/1950他们正年轻/志愿军老兵讲述历史1080_10s_5_test”
output_dir_2 = “/disk2/sunyingli/1950他们正年轻/志愿军老兵讲述历史4K_10s_5_test”

设置截取间隔（以秒为单位）

interval = 10

设置需要截取的间隔次数

num_intervals = 100000

设置每个间隔中需要截取的帧数

frames_per_interval = 5

调用函数进行截取帧

extract_frames(input_file, input_file_1, input_file_2, output_dir, output_dir_1, output_dir_2, interval, num_intervals, frames_per_interval)