网页视频抓取脚本( 2020年02月21日15:50:04)

优采云 发布时间: 2021-11-13 12:11

  网页视频抓取脚本(

2020年02月21日15:50:04)

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

  更新时间:2020年2月21日15:50:04 作者:anoyi

  本文文章主要介绍python ffmpeg任意提取视频帧的方法。文章通过示例代码进行了详细介绍,对大家的学习或工作具有一定的参考学习价值。有需要的朋友可以关注下方小编一起学习

  环境准备

  1、安装FFmpeg

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

  2、安装ffmpeg-python

  

pip3 install ffmpeg-python

  3、[可选] 安装 opencv-python

  

pip3 install opencv-python

  4、[可选] 安装 numpy

  

pip3 install numpy

  视频帧提取

  准备视频素材

  抖音视频素材下载:

  根据视频帧数提取任意帧

  

import ffmpeg

import numpy

import cv2

import sys

import random

def 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 out

def 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 ffmpeg

import numpy

import cv2

import sys

import random

def 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 out

def 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()

  相关信息

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

0 个评论

要回复文章请先登录注册


官方客服QQ群

微信人工客服

QQ人工客服


线