ComfyUI InfoComfyUI Info

ComfyUI-Impact-Subpack

ComfyUI-Impact-Subpack提供了基于 Ultralytics 的检测功能,支持边界框与分割模型的加载与管理,增强图像分析与细节处理能力。

ltdrdata

v1.3.2
203 stars
GitHub

Description

ComfyUI-Impact-Subpack 节点包

该扩展包提供了专门设计的节点,用于与 ComfyUI Impact Pack 配合使用时增强功能,包括 UltralyticsDetectorProvider 功能。

主要特性

核心节点

  • UltralyticsDetectorProvider - 初始化 Ultralytics 模型以生成 SEGM_DETECTOR 和 BBOX_DETECTOR 输出
  • MMDetDetectorProvider 不同,它为分割模型提供 BBOX_DETECTOR 输出
  • 通过 ComfyUI-Manager 的下载界面访问多个预训练模型

模型管理

Ultralytics 模型实现

  • 按类型在专用目录中组织模型:
    • 边界框模型:models/ultralytics/bbox
    • 分割模型:models/ultralytics/segm
  • 通过以下渠道获取模型:

配置设置

路径自定义

extra_model_paths.yaml 中添加以下条目:

  • ultralytics_bbox: YOLO 边界框模型路径
  • ultralytics_segm: YOLO 分割模型路径
  • ultralytics: 支持子目录(bbox/segm/

安全协议

  • PyTorch 2.6+ 实现了安全模型加载(weights_only),防止恶意代码执行
  • <user_directory>/default/ComfyUI-Impact-Subpack/model-whitelist.txt 中将可信模型加入白名单以绕过限制
    • 注意:<user_directory> 通常位于 ComfyUI/user 目录下

安装方法

推荐方式:ComfyUI-Manager

  • 在 ComfyUI-Manager 中搜索 "ComfyUI Impact Subpack" 并选择“安装”

替代方式:手动安装

  1. 进入目录:cd custom_nodes
  2. 克隆仓库:git clone https://github.com/ltdrdata/ComfyUI-Impact-Subpack
  3. 进入目录:cd ComfyUI-Impact-Subpack
  4. 安装依赖:pip install -r requirements.txt
    • 关键说明:
      • 必须使用 ComfyUI 自带的 Python 环境
      • 可移植版本需使用 <安装路径>\python_embeded\python.exe -m pip
      • 对于虚拟环境,请先激活再使用 pip
  5. 重启 ComfyUI

致谢

核心技术

  • ComfyUI - 模块化 Stable Diffusion 接口

模型资源