中文资料找半天没找到,最后google 英文到stackoverflow 才找到,记录下
先安装好ffmpeg,网上教程可查
然后opencv等库

import os
import cv2
import subprocess 

filename = r'视频文件路径' 


def get_frame_types(video_fn):
    command = 'ffprobe -v error -show_entries frame=pict_type -of default=noprint_wrappers=1'.split()
    out = subprocess.check_output(command + [video_fn]).decode()
    frame_types = out.replace('pict_type=', '').split()
    return zip(range(len(frame_types)), frame_types)


from PIL import Image


def save_i_keyframes(video_fn):
    frame_types = get_frame_types(video_fn)
    i_frames = [x[0] for x in frame_types if x[1] == 'I']
    if i_frames:
        basename = os.path.splitext(os.path.basename(video_fn))[0]
        res_dir = os.path.join(os.path.dirname(video_fn), 'frame')
        os.mkdir(res_dir)
        cap = cv2.VideoCapture(video_fn)
        for frame_no in i_frames:
            cap.set(cv2.CAP_PROP_POS_FRAMES, frame_no)
            ret, frame = cap.read()
            outname = basename + '_i_frame_' + str(frame_no) + '.jpg'
            Image.fromarray(frame).save(os.path.join(res_dir, outname))
            print('Saved: ' + outname)
        cap.release()
    else:
        print('No I-frames in ' + video_fn)


if __name__ == '__main__':
    save_i_keyframes(filename)
Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐