如何通过MoviePy解决大量内存使用问题?

时间:2015-05-02 08:20:05

标签: python memory kill moviepy

我正在尝试使用MoviePy从大量图像制作视频。该方法适用于少量图像,但该过程因大量图像而被杀死。在添加了大约500个图像时,Python进程使用了​​大约一半可用的易失性内存。还有更多的图像。

我该如何解决这个问题?我希望处理完成,我不介意处理是否需要更长时间,但如果我能以某种方式限制内存和CPU使用率,那将会很好。使用当前的方法,机器在处理时变得几乎无法使用。

代码如下:

import os
import time
from   moviepy.editor import *

def ls_files(
    path = "."
    ):
    return([fileName for fileName in os.listdir(path) if os.path.isfile(
        os.path.join(path, fileName)
    )])

def main():

    listOfFiles = ls_files()
    listOfTileImageFiles = [fileName for fileName in listOfFiles \
        if "_tile.png" in fileName
    ]
    numberOfTiledImages = len(listOfTileImageFiles)

    # Create a video clip for each image.
    print("create video")
    videoClips = []
    imageDurations = []
    for imageNumber in range(0, numberOfTiledImages):
        imageFileName = str(imageNumber) + "_tile.png"
        print("add image {fileName}".format(
            fileName = imageFileName
        ))
        imageClip = ImageClip(imageFileName)
        duration  = 0.1
        videoClip = imageClip.set_duration(duration)
        # Determine the image start time by calculating the sum of the durations
        # of all previous images.
        if imageNumber != 0:
            videoStartTime = sum(imageDurations[0:imageNumber])
        else:
            videoStartTime = 0
        videoClip = videoClip.set_start(videoStartTime)
        videoClips.append(videoClip)
        imageDurations.append(duration)
    fullDuration = sum(imageDurations)
    video = concatenate(videoClips)
    video.write_videofile(
        "video.mp4",
        fps         = 30,
        codec       = "mpeg4",
        audio_codec = "libvorbis"
    )

if __name__ == "__main__":
    main()

1 个答案:

答案 0 :(得分:3)

如果我理解正确,您希望将不同的图像用作视频的帧。

在这种情况下,您应该使用ImageSequenceClip()(它在库中,但尚未在网络文档中,请参阅here for the doc)。

基本上,你只需写

clip = ImageSequenceClip("some/directory/", fps=10)
clip.write_videofile("result.mp4")

它将以字母数字顺序读取目录中的图像,同时在内存中一次只保留一帧。

当我在这里时,您还可以向ImageSequenceClip提供文件名列表或numpy数组列表。

请注意,如果您只想将图片转换为视频,而不想添加标题或与其他视频合成等其他任何内容,那么您可以直接使用ffmpeg进行操作。从内存中命令应该是:

ffmpeg -i *_tile.png -r 10 -o result.mp4