管道PIL图像到ffmpeg stdin - Python

时间:2017-04-27 07:06:33

标签: python ffmpeg phantomjs python-imaging-library

我试图将html5视频转换为mp4视频,并且是通过PhantomJS随时间进行屏幕拍摄来实现的

我还使用PIL裁剪图像,所以最终我的代码大致是:

while time() < end_time:
    screenshot_list.append(phantom.get_screenshot_as_base64())
.
.
for screenshot in screenshot_list:
    im = Image.open(BytesIO(base64.b64decode(screenshot)))
    im = im.crop((left, top, right, bottom))

现在我正在保存以删除所有这些图像并使用保存文件中的ffmpeg:

os.system('ffmpeg -r {fps} -f image2 -s {width}x{height} -i {screenshots_dir}%04d.png -vf scale={width}:-2 '
      '-vcodec libx264 -crf 25 -vb 20M -pix_fmt yuv420p {output}'.format(fps=fps, width=width,
                                                                  screenshots_dir=screenshots_dir,
                                                                  height=height, output=output))

但是我想要而不是使用那些保存的文件,能够将PIL.Images直接传递给ffmpeg,我该怎么做?

1 个答案:

答案 0 :(得分:9)

赏金消失了,但我找到了解决方案。

将所有屏幕截图作为base64字符串获取后,我将它们写入带有以下代码的子进程

import subprocess as sp

# Generating all of the screenshots as base64 
# in a variable called screenshot_list

cmd_out = ['ffmpeg',
           '-f', 'image2pipe',
           '-vcodec', 'png',
           '-r', '30',  # FPS 
           '-i', '-',  # Indicated input comes from pipe 
           '-vcodec', 'png',
           '-qscale', '0',
           '/home/user1/output_dir/video.mp4']

pipe = sp.Popen(cmd_out, stdin=sp.PIPE)

for screenshot in screenshot_list:
    im = Image.open(BytesIO(base64.b64decode(screenshot)))
    im.save(pipe.stdin, 'PNG')

pipe.stdin.close()
pipe.wait()

# Make sure all went well
if pipe.returncode != 0:
    raise sp.CalledProcessError(pipe.returncode, cmd_out)

如果执行时间有问题,您可以将图像保存为JPEG,并使用适当的编解码器,但我设法达到的最高质量是使用这些设置