Files
2024-07-04 11:34:32 +08:00

4.0 KiB

Documentation

ReActorMaskHelper节点旨在辅助ReActor框架内的遮罩处理过程。它主要用于组合、膨胀和优化遮罩,以便在人脸交换或修复任务中使用。该节点利用特定的检测提示和遮罩操作技术来优化遮罩处理过程,从而获得更好的结果。

Input types

Required

  • image
    • 'image'参数是遮罩相关操作的主要输入,作为检测、操作和转换过程的基础。
    • Comfy dtype: IMAGE
    • Python dtype: torch.Tensor
  • swapped_image
    • 此参数表示人脸交换操作后的图像,可能需要进一步的遮罩处理或调整。
    • Comfy dtype: IMAGE
    • Python dtype: torch.Tensor
  • bbox_model_name
    • 指定用于边界框检测的模型名称,影响遮罩生成的初始步骤。
    • Comfy dtype: COMBO[STRING]
    • Python dtype: str
  • bbox_threshold
    • 边界框检测的阈值,决定模型在识别潜在兴趣区域时的敏感度。
    • Comfy dtype: FLOAT
    • Python dtype: float
  • bbox_dilation
    • 表示应用于边界框的膨胀因子,可能会扩大考虑遮罩应用的区域。
    • Comfy dtype: INT
    • Python dtype: int
  • bbox_crop_factor
    • 定义在检测到的边界框周围裁剪的程度,影响最终遮罩的覆盖区域。
    • Comfy dtype: FLOAT
    • Python dtype: float
  • bbox_drop_size
    • 边界框被考虑的最小尺寸,过滤掉太小的检测结果。
    • Comfy dtype: INT
    • Python dtype: int
  • sam_model_name
    • 用于SAM(语义注意力遮罩)预测的模型名称,对于优化遮罩细节至关重要。
    • Comfy dtype: COMBO[STRING]
    • Python dtype: str
  • sam_dilation
    • 专门用于SAM预测的膨胀因子,调整遮罩边界以获得更好的拟合或覆盖。
    • Comfy dtype: INT
    • Python dtype: int
  • sam_threshold
    • SAM预测的阈值,根据置信度分数决定包含哪些遮罩。
    • Comfy dtype: FLOAT
    • Python dtype: float
  • bbox_expansion
    • 指定边界框超出其检测边界的扩展程度,影响遮罩覆盖的区域。
    • Comfy dtype: INT
    • Python dtype: int
  • mask_hint_threshold
    • 生成遮罩提示的阈值,用于需要预先识别遮罩区域的过程。
    • Comfy dtype: FLOAT
    • Python dtype: float
  • mask_hint_use_negative
    • 表示是否在遮罩生成过程中使用负面提示(不需遮罩的区域)。
    • Comfy dtype: COMBO[STRING]
    • Python dtype: bool
  • morphology_operation
    • 指定应用于遮罩的形态学操作类型(如膨胀、腐蚀),影响其形状和边界。
    • Comfy dtype: COMBO[STRING]
    • Python dtype: str
  • morphology_distance
    • 形态学操作的距离参数,决定变换的程度。
    • Comfy dtype: INT
    • Python dtype: int
  • blur_radius
    • 遮罩模糊的半径,用于软化边缘或使遮罩与图像更无缝地融合。
    • Comfy dtype: INT
    • Python dtype: int
  • sigma_factor
    • 计算高斯模糊中sigma值的因子,影响遮罩边缘的平滑度。
    • Comfy dtype: FLOAT
    • Python dtype: float

Optional

  • mask_optional
    • 可选的遮罩输入,可用于额外的遮罩操作或调整。
    • Comfy dtype: MASK
    • Python dtype: torch.Tensor

Output types

  • IMAGE
    • 应用遮罩修改后的处理图像,可用于进一步使用或显示。
    • Comfy dtype: IMAGE
    • Python dtype: torch.Tensor
  • MASK
    • 节点生成或修改的最终遮罩,适用于遮罩或合成操作。
    • Comfy dtype: MASK
    • Python dtype: torch.Tensor
  • MASK_PREVIEW
    • 遮罩的预览,通常用于可视化或调试目的。
    • Comfy dtype: IMAGE
    • Python dtype: torch.Tensor
  • SWAPPED_FACE
    • 人脸交换操作后的图像,可能包括遮罩调整以改善整合效果。
    • Comfy dtype: IMAGE
    • Python dtype: torch.Tensor

Usage tips

  • Infra type: CPU
  • Common nodes: unknown

Source code

# Built-in or C extension class, unable to automatically detect source code