import cv2
import os
def extract_frames(video_path, frames_path):
# 打开视频文件
cap = cv2.VideoCapture(video_path)
# 获取视频帧速率
fps = cap.get(cv2.CAP_PROP_FPS)
# 从视频中读取每一帧
success, image = cap.read()
count = 0
# 确保frames目录存在
if not os.path.exists(frames_path):
os.makedirs(frames_path)
# 循环直到视频结束
while success:
# 保存帧为JPG图像
cv2.imwrite(os.path.join(frames_path, f"{count}.jpg"), image)
count += 1
# 读取下一帧
success, image = cap.read()
# 关闭视频文件
cap.release()
print(f"Extracted {count} frames at {fps}fps to {frames_path}")
# 测试
video_path = "E:/test/1.mp4"#需自行修改
frames_path = "E:/test/input"#需自行修改
extract_frames(video_path, frames_path)