使用Python获取视频中的I帧列表

pip install ffmpeg-python

import ffmpeg

def get_iframe_list(video_path):
    try:
        # 打开视频文件
        probe = ffmpeg.probe(video_path)
        video_stream = next((stream for stream in probe['streams'] if stream['codec_type'] == 'video'), None)
        
        if video_stream is None:
            raise ValueError("Could not find video stream in the input file")
        
        # 获取视频帧率
        fps = int(video_stream['avg_frame_rate'].split('/')[0])
        
        # 使用FFmpeg命令获取I帧列表
        command = [
            'ffmpeg',
            '-i', video_path,
            '-vf', f"select=eq(pict_type\,I)",
            '-vsync', 'vfr',
            'frames-%04d.jpeg'
        ]
        
        output = ffmpeg.run(command, capture_stdout=True, capture_stderr=True)
        
        # 解析FFmpeg输出获取I帧时间戳
        iframe_times = []
        for line in output.stderr.decode('utf-8').split('\n'):
            if 'frame=' in line:
                time_str = line.split('time=')[1].split()[0]
                iframe_times.append(time_str)
        
        return iframe_times
    
    except ffmpeg.Error as e:
        print('stdout:', e.stdout.decode('utf-8'))
        print('stderr:', e.stderr.decode('utf-8'))
        raise

# 示例使用
video_path = 'path_to_your_video.mp4'
iframe_list = get_iframe_list(video_path)
print("I帧列表:", iframe_list)