确定视频比特率 Python

问题描述

我有一个从 Telegram 下载的视频,我需要确定它的比特率。 我有moviepy(pip install moviepy,不是开发者版本)。 另外,我有ffmpeg,但我不知道如何在python中使用它。 此外,任何其他图书馆都适合我。

解决方法

这是使用 FFprobe 的解决方案:

  • 执行ffprobe(命令行工具)作为子进程并读取stdout的内容。
    使用参数 -print_format json 获取 JSON 格式的输出。
    要仅获取 bit_rate 条目,请添加参数 -show_entries stream=bit_rate
  • 使用 dict = json.loads(data) 将返回的字符串转换为字典。
  • 从字典中获取比特率并将其转换为 intbit_rate = int(dict['streams'][0]['bit_rate'])

代码示例创建用于测试的示例视频文件(使用 FFmpeg),并获取比特率(使用 FFprobe):

import subprocess as sp
import shlex
import json

input_file_name = 'test.mp4'

# Build synthetic video for testing:
################################################################################
sp.run(shlex.split(f'ffmpeg -y -f lavfi -i testsrc=size=320x240:rate=30 -f lavfi -i sine=frequency=400 -f lavfi -i sine=frequency=1000 -filter_complex amerge -vcodec libx264 -crf 17 -pix_fmt yuv420p -acodec aac -ar 22050 -t 10 {input_file_name}'))
################################################################################

# Use FFprobe for 
# Execute ffprobe (to get specific stream entries),and get the output in JSON format
data = sp.run(shlex.split(f'ffprobe -v error -select_streams v:0 -show_entries stream=bit_rate -print_format json {input_file_name}'),stdout=sp.PIPE).stdout
dict = json.loads(data)  # Convert data from JSON string to dictionary
bit_rate = int(dict['streams'][0]['bit_rate'])  # Get the bitrate.

print(f'bit_rate = {bit_rate}')

注意事项:

  • 对于 MKV 等某些视频容器,没有 bit_rate 信息,因此需要不同的解决方案。
  • 代码示例假定 ffmpeg 和 ffprobe(命令行工具)在执行路径中。

没有 bit_rate 信息的容器(如 MKV)的解决方案:

基于以下post,我们可以对所有视频数据包的大小求和。
我们还可以总结所有数据包的持续时间。
平均比特率等于:total_size_in_bits / total_duration_in_seconds

这里是计算 MKV 视频文件平均比特率的代码示例:

import subprocess as sp
import shlex
import json

input_file_name = 'test.mkv'

# Build synthetic video for testing (MKV video container):
################################################################################
sp.run(shlex.split(f'ffmpeg -y -f lavfi -i testsrc=size=320x240:rate=30 -f lavfi -i sine=frequency=400 -f lavfi -i sine=frequency=1000 -filter_complex amerge -vcodec libx264 -crf 17 -pix_fmt yuv420p -acodec aac -ar 22050 -t 10 {input_file_name}'))
################################################################################

# https://superuser.com/questions/1106343/determine-video-bitrate-using-ffmpeg
# Calculating the bitrate by summing all lines except the last one,and dividing by the value in the last line.
data = sp.run(shlex.split(f'ffprobe -select_streams v:0 -show_entries packet=size,duration -of compact=p=0:nk=1 -print_format json {input_file_name}'),stdout=sp.PIPE).stdout
dict = json.loads(data)  # Convert data from JSON string to dictionary

# Sum total packets size and total packets duration.
sum_packets_size = 0
sum_packets_duration = 0
for p in dict['packets']:
    sum_packets_size += float(p['size'])  # Sum all the packets sizes (in bytes)
    sum_packets_duration += float(p['duration'])  # Sum all the packets durations (in mili-seconds).

# bitrate is the total_size / total_duration (multiply by 1000 because duration is in msec units,and by 8 for converting from bytes to bits).
bit_rate = (sum_packets_size / sum_packets_duration) * 8*1000

print(f'bit_rate = {bit_rate}')
,
import moviepy.editor as mp

video = mp.VideoFileClip('vid.mp4')
mp3 = video.audio
if mp3 is not None:
    mp3.write_audiofile("vid_audio.mp3")
mp3_size =  os.path.getsize("vid_audio.mp3")
vid_size = os.path.getsize('vid.mp4')
duration = video.duration


bitrate = int((((vid_size - mp3_size)/duration)/1024*8))

相关问答

Selenium Web驱动程序和Java。元素在(x,y)点处不可单击。其...
Python-如何使用点“。” 访问字典成员?
Java 字符串是不可变的。到底是什么意思?
Java中的“ final”关键字如何工作?(我仍然可以修改对象。...
“loop:”在Java代码中。这是什么,为什么要编译?
java.lang.ClassNotFoundException:sun.jdbc.odbc.JdbcOdbc...