CavinHuang-comfyui-nodes-docs/docs/LoadInstancePositionNetModel.md

Documentation

• Class name: LoadInstancePositionNetModel
• Category: instance/loaders
• Output node: False
• Repo Ref: https://github.com/comfyanonymous/ComfyUI

LoadInstancePositionNetModel节点用于加载PositionNet模型，这是实例扩散框架的一个重要组成部分。该节点通过读取指定的模型文件并应用必要的配置来实现PositionNet的加载。它使用预定义的参数初始化PositionNet，并提供了包含分割信息的选项，从而为后续的基于实例的处理任务做好准备。

Input types

Required

• model_filename
  • 指定要加载的PositionNet模型文件名。这个参数对于从预定义目录中识别和检索正确的模型文件至关重要。
  • Comfy dtype: COMBO[STRING]
  • Python dtype: str
• use_segs
  • 决定是否在PositionNet模型配置中使用分割信息。这个布尔标志允许根据分割数据的存在与否灵活地调整模型。
  • Comfy dtype: BOOLEAN
  • Python dtype: bool

Output types

• positionnet
  • 加载完成的PositionNet模型，可用于基于实例的处理任务。这个输出封装了使用指定配置初始化的模型。
  • Comfy dtype: POSITIONNET
  • Python dtype: torch.nn.Module
• fusers
  • 加载完成的fusers模型，它是实例扩散框架的一部分，准备与其他组件集成。
  • Comfy dtype: FUSERS
  • Python dtype: torch.nn.Module
• scaleu
  • 加载完成的ScaleU模型，实例扩散框架的另一个组件，为实例的缩放和处理做好准备。
  • Comfy dtype: SCALEU
  • Python dtype: torch.nn.Module

Usage tips

• Infra type: GPU
• Common nodes: unknown

Source code

class LoadInstancePositionNetNode:
    @classmethod
    def INPUT_TYPES(s):
        return {"required": {
            "model_filename": (get_model_list(constants.INSTANCE_POSITIONNET_DIR),),
            "use_segs": ("BOOLEAN", {"default": True}),
        }}

    RETURN_TYPES = ("POSITIONNET", "FUSERS", "SCALEU",)
    FUNCTION = "load_model"

    CATEGORY = "instance/loaders"

    def load_model(self, model_filename: str, use_segs: bool):
        checkpoint = load_checkpoint(
            constants.INSTANCE_POSITIONNET_DIR, model_filename)
        params = get_positionnet_default_params()
        params["use_segs"] = use_segs
        model = prepare_positionnet(checkpoint, params)
        positionnet = {
            'model': model,
        }
        return (positionnet,)