import cv2  
import os  

def extract_frame(video_path, frame_number, output_path):  
    # 打开视频文件  
    cap = cv2.VideoCapture(video_path)  

    if not cap.isOpened():  
        print(f"无法打开视频文件: {video_path}")  
        return  

    # 设置视频捕捉的位置到指定帧  
    cap.set(cv2.CAP_PROP_POS_FRAMES, frame_number)  

    # 读取指定的帧  
    success, frame = cap.read()  

    if success:  
        # 保存帧为指定路径的文件  
        cv2.imwrite(output_path, frame)  
        print(f"已成功提取帧 {frame_number} 并保存为 {output_path}")  
    else:  
        print(f"未能读取帧 {frame_number}。请检查帧编号是否超出范围。")  

    # 释放资源  
    cap.release()  

if __name__ == "__main__":  
    video_file = "/mnt/lpai-dione/ssai/cvg/team/wjj/ViViD/dataset/ViViD/dresses/densepose/803137_detail.mp4"  # 替换为你的 MP4 文件路径  
    frame_to_extract = 24  # 需要提取的帧编号  
    output_file = "/mnt/lpai-dione/ssai/cvg/team/wjj/ViViD/configs/valid/densepose_images/803137_in_xl.jpg"  # 替换为你想保存的路径  

    # 创建包含输出文件的目录（如果不存在）  
    output_dir = os.path.dirname(output_file)  
    if not os.path.exists(output_dir) and output_dir:  
        os.makedirs(output_dir)  

    extract_frame(video_file, frame_to_extract, output_file)