你可以"流"图像到ffmpeg构建一个视频,而不是将它们保存到磁盘？

Question

我最近的工作涉及以编程方式制作视频.在python中,典型的工作流程看起来像这样:

import subprocess, Image, ImageDraw

for i in range(frames_per_second * video_duration_seconds):
    img = createFrame(i)
    img.save("%07d.png" % i)

subprocess.call(["ffmpeg","-y","-r",str(frames_per_second),"-i", "%07d.png","-vcodec","mpeg4", "-qscale","5", "-r", str(frames_per_second), "video.avi"])

此工作流程为视频中的每个帧创建一个图像并将其保存到磁盘.保存所有图像后,调用ffmpeg以构建来自所有图像的视频.

将图像保存到磁盘(而不是在内存中创建图像)会占用此处的大部分周期,并且似乎不需要.有没有办法执行相同的功能,但没有将图像保存到磁盘？因此,将调用ffmpeg并构建图像并在构造后立即将其提供给ffmpeg.

Answer 1

好的,我得到了它的工作.感谢LordNeckbeard建议使用image2pipe.我不得不使用jpg编码而不是png,因为带有png的image2pipe对我的ffmpeg的验证不起作用.第一个脚本与你的问题代码基本相同,只是我实现了一个简单的图像创建,只创建从黑色到红色的图像.我还添加了一些代码来计算执行时间.

串行执行

import subprocess, Image

fps, duration = 24, 100
for i in range(fps * duration):
    im = Image.new("RGB", (300, 300), (i, 1, 1))
    im.save("%07d.jpg" % i)
subprocess.call(["ffmpeg","-y","-r",str(fps),"-i", "%07d.jpg","-vcodec","mpeg4", "-qscale","5", "-r", str(fps), "video.avi"])

并行执行(没有图像保存到磁盘)

import Image
from subprocess import Popen, PIPE

fps, duration = 24, 100
p = Popen(['ffmpeg', '-y', '-f', 'image2pipe', '-vcodec', 'mjpeg', '-r', '24', '-i', '-', '-vcodec', 'mpeg4', '-qscale', '5', '-r', '24', 'video.avi'], stdin=PIPE)
for i in range(fps * duration):
    im = Image.new("RGB", (300, 300), (i, 1, 1))
    im.save(p.stdin, 'JPEG')
p.stdin.close()
p.wait()

结果很有意思,我运行每个脚本3次以比较性能: serial:

12.9062321186
12.8965060711
12.9360799789

平行:

8.67797684669
8.57139396667
8.38926696777

所以似乎并行版本的速度提高了约1.5倍.

Answer 2

imageio直接支持这一点。它使用 FFMPEG 和Video Acceleration API，速度非常快：

import imageio

writer = imageio.get_writer('video.avi', fps=fps)
for i in range(frames_per_second * video_duration_seconds):
    img = createFrame(i)
    writer.append_data(img)
writer.close()