import os
import glob
import cv2
def extract_frames_from_video(input_video_path, output_folder, frame_interval=10):
video_capture = cv2.VideoCapture(input_video_path)
frame_count = int(video_capture.get(cv2.CAP_PROP_FRAME_COUNT))
fps = int(video_capture.get(cv2.CAP_PROP_FPS))
if not video_capture.isOpened():
print(f"Error: Cannot open video file {input_video_path}.")
return
current_frame = 0
frame_number = 0
while current_frame < frame_count:
ret, frame = video_capture.read()
if not ret:
print(f"Error: Failed to read frame from {input_video_path}.")
break
if current_frame % frame_interval == 0:
#output_file_path = os.path.join(output_folder, f"{os.path.splitext(os.path.basename(input_video_path))[0]}_frame_{frame_number:04d}.jpg")
output_file_path = os.path.join(output_folder, f"{os.path.splitext(os.path.basename(input_video_path))[0]}_{frame_number:02d}.jpg")
cv2.imwrite(output_file_path, frame)
frame_number += 1
current_frame += 1
video_capture.release()
print(f"Frames extracted from {input_video_path}: {frame_number}")
def process_all_videos_in_folder(input_folder, output_folder, frame_interval=10):
os.makedirs(output_folder, exist_ok=True)
video_files = glob.glob(os.path.join(input_folder, "*.avi"))
for video_file in video_files:
extract_frames_from_video(video_file, output_folder, frame_interval)
# 使用示例:
input_folder = "D:\\dataset\\YawDD dataset\\Mirror\\Male" # 包含多个AVI视频的文件夹路径
output_folder = "D:\\dataset\\YawDD dataset\\Mirror\\Male_pic" # 存放提取的图像的文件夹路径
frame_interval = 50 # 每隔多少帧提取一张图像
process_all_videos_in_folder(input_folder, output_folder, frame_interval)
Python数据处理——按照帧率从MP4格式视频中切出图片
最新推荐文章于 2024-05-24 12:43:43 发布