CavinHuang-comfyui-nodes-docs/docs/Create Video from Path.md

Documentation

• Class name: Create Video from Path
• Category: WAS Suite/Animation
• Output node: False

该节点能够从一系列图像创建视频，支持自定义过渡效果、帧延迟以及视频属性如帧率和分辨率。它利用计算机视觉技术在图像之间生成平滑过渡，非常适合创建变形效果或幻灯片。

Input types

Required

• transition_frames
  • 指定用于图像间过渡的帧数，用于实现平滑的变形效果。
  • Comfy dtype: INT
  • Python dtype: int
• image_delay_sec
  • 决定每张图像在过渡到下一张之前的延迟时间（以秒为单位），用于控制视频中每张图像的显示时长。
  • Comfy dtype: FLOAT
  • Python dtype: int
• fps
  • 设置输出视频的帧率（fps），影响视频的流畅度和播放速度。
  • Comfy dtype: INT
  • Python dtype: int
• max_size
  • 定义图像宽度或高度的最大尺寸（以像素为单位），确保视频保持一致的分辨率。
  • Comfy dtype: INT
  • Python dtype: int
• input_path
  • 输入图像所在的目录路径。
  • Comfy dtype: STRING
  • Python dtype: str
• output_path
  • 输出视频将保存的目录路径。
  • Comfy dtype: STRING
  • Python dtype: str
• filename
  • 输出视频文件的名称。
  • Comfy dtype: STRING
  • Python dtype: str
• codec
  • 用于编码视频的编解码器，影响兼容性和质量。
  • Comfy dtype: COMBO[STRING]
  • Python dtype: str

Output types

• filepath_text
  • Comfy dtype: STRING
  • 创建的视频文件的路径。
  • Python dtype: str
• filename_text
  • Comfy dtype: STRING
  • 创建的视频文件的名称，用于确认输出。
  • Python dtype: str

Usage tips

• Infra type: GPU
• Common nodes: unknown

Source code

class WAS_Create_Video_From_Path:
    def __init__(self):
        pass

    @classmethod
    def INPUT_TYPES(cls):
        WTools = WAS_Tools_Class()
        v = WTools.VideoWriter()
        codecs = []
        for codec in v.get_codecs():
            codecs.append(codec.upper())
        codecs = sorted(codecs)
        return {
            "required": {
                "transition_frames": ("INT", {"default":30, "min":0, "max":120, "step":1}),
                "image_delay_sec": ("FLOAT", {"default":2.5, "min":0.01, "max":60000.0, "step":0.01}),
                "fps": ("INT", {"default":30, "min":1, "max":60.0, "step":1}),
                "max_size": ("INT", {"default":512, "min":128, "max":1920, "step":1}),
                "input_path": ("STRING", {"default": "./ComfyUI/input", "multiline": False}),
                "output_path": ("STRING", {"default": "./ComfyUI/output", "multiline": False}),
                "filename": ("STRING", {"default": "comfy_video", "multiline": False}),
                "codec": (codecs,),
            }
        }

    @classmethod
    def IS_CHANGED(cls, **kwargs):
        return float("NaN")

    RETURN_TYPES = (TEXT_TYPE,TEXT_TYPE)
    RETURN_NAMES = ("filepath_text","filename_text")
    FUNCTION = "create_video_from_path"

    CATEGORY = "WAS Suite/Animation"

    def create_video_from_path(self, transition_frames=10, image_delay_sec=10, fps=30, max_size=512,
                            input_path="./ComfyUI/input", output_path="./ComfyUI/output", filename="morph", codec="H264"):

        conf = getSuiteConfig()
        if not conf.__contains__('ffmpeg_bin_path'):
            cstr(f"Unable to use MP4 Writer because the `ffmpeg_bin_path` is not set in `{WAS_CONFIG_FILE}`").error.print()
            return ("","")

        if conf.__contains__('ffmpeg_bin_path'):
            if conf['ffmpeg_bin_path'] != "/path/to/ffmpeg":
                sys.path.append(conf['ffmpeg_bin_path'])
                os.environ["OPENCV_FFMPEG_CAPTURE_OPTIONS"] = "rtsp_transport;udp"
                os.environ['OPENCV_FFMPEG_BINARY'] = conf['ffmpeg_bin_path']

        if output_path.strip() in [None, "", "."]:
            output_path = "./ComfyUI/output"

        if transition_frames < 0:
            transition_frames = 0
        elif transition_frames > 60:
            transition_frames = 60

        if fps < 1:
            fps = 1
        elif fps > 60:
            fps = 60

        tokens = TextTokens()

        # Check if output_path is an absolute path
        if not os.path.isabs(output_path):
            output_path = os.path.abspath(os.path.join(*tokens.parseTokens(output_path).split('/')))

        output_file = os.path.join(output_path, tokens.parseTokens(filename))

        if not os.path.exists(output_path):
            os.makedirs(output_path, exist_ok=True)

        WTools = WAS_Tools_Class()
        MP4Writer = WTools.VideoWriter(int(transition_frames), int(fps), int(image_delay_sec), max_size, codec)
        path = MP4Writer.create_video(input_path, output_file)

        return (path, filename)