Python moviepy.editor方法代码示例

# 需要导入模块: import moviepy [as 别名]
# 或者: from moviepy import editor [as 别名]
def npy_to_video(npy, filename, fps=10, preview=True, convert='gif'):
    """Convert a numpy array into a gif file at the location specified by filename.

        # Arguments

        convert: Default empty string is no conversion, options are gif and mp4.
        preview: pop open a preview window to view the video data.
    """
    # Useful moviepy instructions https://github.com/Zulko/moviepy/issues/159
    # TODO(ahundt) currently importing moviepy prevents python from exiting. Once this is resolved remove the import below.
    import moviepy.editor as mpy
    clip = mpy.ImageSequenceClip(list(npy), fps)
    if preview:
        # https://stackoverflow.com/a/41771413
        clip.preview()
    if convert == 'gif':
        clip.write_gif(filename)
    elif convert:
        clip.write_videofile(filename) 

# 需要导入模块: import moviepy [as 别名]
# 或者: from moviepy import editor [as 别名]
def test_raises_error_if_moviepy_not_available(self):
        # OK, so this is a nasty nasty test with import fuckery going on.
        try:
            # First we make moviepy.editor unimportable
            sys.modules["moviepy.editor"] = None
            # but we have already imported torchvideo.tools so that code is
            # cached with a reference to the *original* moviepy, so we need to reload
            # it to invoke the failure path of the import code
            importlib.reload(torchvideo.tools)
            from torchvideo.tools import show_video

            # Now we've imported torchvideo.tools where moviepy.editor was
            # *not* available.
            with pytest.raises(ImportError, match="please install moviepy"):
                show_video(self.frames)
        finally:
            # We've borked up the system modules and we have to put them right otherwise
            # any tests that depend on torchvideo.tools or moviepy.editor are
            # going to fail.
            # We first restore the original moviepy editor object back
            sys.modules["moviepy.editor"] = original_moviepy_editor
            # I don't think we need to reload moviepy (other tests pass without this
            # line), but for good measure let's reload it just in case it does
            # something useful.
            importlib.reload(moviepy)
            # We also want to reload torchvideo.tools so moviepy *is* available
            importlib.reload(torchvideo.tools) 

# 需要导入模块: import moviepy [as 别名]
# 或者: from moviepy import editor [as 别名]
def test_uses_ipython_display_if_ipython_is_available(self, monkeypatch):
        display_mock = Mock()
        with monkeypatch.context() as ctx:
            ctx.setattr(moviepy.video.io.html_tools, "ipython_available", True)
            ctx.setattr(
                moviepy.editor.ImageSequenceClip, "ipython_display", display_mock
            )

            importlib.reload(torchvideo.tools)
            show_video(self.frames)

            display_mock.assert_called_once_with() 

# 需要导入模块: import moviepy [as 别名]
# 或者: from moviepy import editor [as 别名]
def test_fallsback_on_pygame_display(self, monkeypatch):
        show_mock = Mock()
        with monkeypatch.context() as ctx:
            ctx.setattr(moviepy.video.io.html_tools, "ipython_available", False)
            ctx.setattr(moviepy.editor.ImageSequenceClip, "show", show_mock)

            importlib.reload(torchvideo.tools)
            show_video(self.frames)

            show_mock.assert_called_once_with()