python ffmpeg提取关键帧
中文资料找半天没找到,最后google 英文到stackoverflow 才找到,记录下先安装好ffmpeg,网上教程可查然后opencv等库import osimport cv2import subprocessfilename = r'视频文件路径'def get_frame_types(video_fn):command = 'ffprobe -v error -show_entries f
·
中文资料找半天没找到,最后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)
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐



所有评论(0)