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)