142 lines
4.8 KiB
Markdown
142 lines
4.8 KiB
Markdown
|
|
# 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
|
|
```python
|
|
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)
|
|
|
|
```
|