Question

我正在尝试获取Python视频中所有I帧的索引列表（以后将它们的一部分另存为JPEG）。现在，我可以在终端中使用FFProbe遍历所有帧，看看哪个是I帧：

ffprobe -select_streams v -show_frames -show_entries frame=pict_type -of csv 18s.mp4

这给了我这样的东西：

frame,I
frame,B
frame,P
frame,B
frame,P
frame,B

但是我该如何在Python（我在Windows中）中做到这一点并获取其所有索引的列表？

Answer 1

您可能让FFmpeg只是将i帧输出为JPG。并使用python包装器触发此命令。

这会将所有i帧输出为JPG图像。

ffmpeg -i 2.flv -vf "select=eq(pict_type\,I)" -vsync vfr frame-%02d.jpg

请在此评论类似的superuser.com问题。 https://superuser.com/questions/669716/how-to-extract-all-key-frames-from-a-video-clip#comment1184347_669733

希望有帮助。干杯。

伊恩

Answer 2

从 here 获得见解，我能够通过ffprobe做到这一点：

def iframes():
    if not os.path.exists(iframe_path):
        os.mkdir(iframe_path)
    command = 'ffprobe -v error -show_entries frame=pict_type -of default=noprint_wrappers=1'.split()
    out = subprocess.check_output(command + [filename]).decode()
    f_types = out.replace('pict_type=','').split()
    frame_types = zip(range(len(f_types)), f_types)
    i_frames = [x[0] for x in frame_types if x[1]=='I']
    if i_frames:
        cap = cv2.VideoCapture(filename)
        for frame_no in i_frames:
            cap.set(cv2.CAP_PROP_POS_FRAMES, frame_no)
            ret, frame = cap.read()
            outname = iframe_path+'i_frame_'+str(frame_no)+'.jpg'
            cv2.imwrite(outname, frame)
        cap.release()
        print("I-Frame selection Done!!")


if __name__ == '__main__':
    iframes()

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

2 个答案: