制作数据集时对视频流抽帧可使用以下代码:
import cv2
import os
def extract_frames(video_path, output_path, frame_rate):
# 创建输出路径
os.makedirs(output_path, exist_ok=True)
# 打开视频文件
video = cv2.VideoCapture(video_path)
# 获取视频的帧率
fps = video.get(cv2.CAP_PROP_FPS)
print(f"获取视频的帧率:{fps}")
# 计算每隔多少帧抽取一帧
frame_interval = int(fps*frame_rate)
# 初始化帧计数器
frame_count = 0
num = 0
while True:
# 读取视频的一帧
ret, frame = video.read()
# 如果无法读取到帧,则退出循环
if not ret:
break
# 如果帧计数器是frame_interval的倍数,则保存该帧
if frame_count % frame_interval == 0:
# 构造保存路径
save_path = os.path.join(output_path, f"frame_{frame_count}.jpg")
# 保存帧为图片
cv2.imwrite(save_path, frame)
num = frame_count/82 #视频fps为25,设定为3.3s抽帧,相乘后82.5取的82,具体看自身情况
print(f"当前处理到第{num}张")
# 增加帧计数器
frame_count += 1
# 释放视频对象
video.release()
# 调用函数进行抽帧处理
video_path = "name.mp4" #修改此处为输入路径
output_path = "path/to/output" #修改此处为输出路径
frame_rate = 3.3 #每多少秒抽帧,此处设定为3.3秒抽取一张图片
extract_frames(video_path, output_path, frame_rate)