python ffmpeg任意提取视频帧的方法

时间:2021-05-22

环境准备

1、安装 FFmpeg

音/视频工具 FFmpeg 简易安装文档

2、安装 ffmpeg-python

pip3 install ffmpeg-python

3、【可选】安装 opencv-python

pip3 install opencv-python

4、【可选】安装 numpy

pip3 install numpy

视频帧提取

准备视频素材

抖音视频素材下载:https://anoyi.com/dy/top

基于视频帧数提取任意一帧

import ffmpegimport numpyimport cv2import sysimport randomdef read_frame_as_jpeg(in_file, frame_num): """ 指定帧数读取任意帧 """ out, err = ( ffmpeg.input(in_file) .filter('select', 'gte(n,{})'.format(frame_num)) .output('pipe:', vframes=1, format='image2', vcodec='mjpeg') .run(capture_stdout=True) ) return outdef get_video_info(in_file): """ 获取视频基本信息 """ try: probe = ffmpeg.probe(in_file) video_stream = next((stream for stream in probe['streams'] if stream['codec_type'] == 'video'), None) if video_stream is None: print('No video stream found', file=sys.stderr) sys.exit(1) return video_stream except ffmpeg.Error as err: print(str(err.stderr, encoding='utf8')) sys.exit(1)if __name__ == '__main__': file_path = '/Users/admin/Downloads/拜无忧.mp4' video_info = get_video_info(file_path) total_frames = int(video_info['nb_frames']) print('总帧数:' + str(total_frames)) random_frame = random.randint(1, total_frames) print('随机帧:' + str(random_frame)) out = read_frame_as_jpeg(file_path, random_frame) image_array = numpy.asarray(bytearray(out), dtype="uint8") image = cv2.imdecode(image_array, cv2.IMREAD_COLOR) cv2.imshow('frame', image) cv2.waitKey()

基于时间提取任意一帧

import ffmpegimport numpyimport cv2import sysimport randomdef read_frame_by_time(in_file, time): """ 指定时间节点读取任意帧 """ out, err = ( ffmpeg.input(in_file, ss=time) .output('pipe:', vframes=1, format='image2', vcodec='mjpeg') .run(capture_stdout=True) ) return outdef get_video_info(in_file): """ 获取视频基本信息 """ try: probe = ffmpeg.probe(in_file) video_stream = next((stream for stream in probe['streams'] if stream['codec_type'] == 'video'), None) if video_stream is None: print('No video stream found', file=sys.stderr) sys.exit(1) return video_stream except ffmpeg.Error as err: print(str(err.stderr, encoding='utf8')) sys.exit(1)if __name__ == '__main__': file_path = '/Users/admin/Downloads/拜无忧.mp4' video_info = get_video_info(file_path) total_duration = video_info['duration'] print('总时间:' + total_duration + 's') random_time = random.randint(1, int(float(total_duration)) - 1) + random.random() print('随机时间:' + str(random_time) + 's') out = read_frame_by_time(file_path, random_time) image_array = numpy.asarray(bytearray(out), dtype="uint8") image = cv2.imdecode(image_array, cv2.IMREAD_COLOR) cv2.imshow('frame', image) cv2.waitKey()

相关资料
https://github.com/kkroening/ffmpeg-python/tree/master/examples

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。

声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。如涉及版权问题,请提交至online#300.cn邮箱联系删除。

相关文章