ComfyUI-Impact-Subpack
ComfyUI-Impact-Subpack提供了基于 Ultralytics 的检测功能,支持边界框与分割模型的加载与管理,增强图像分析与细节处理能力。
ltdrdata
v1.3.2
203 stars
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
- 边界框模型:
- 通过以下渠道获取模型:
- ComfyUI-Manager(搜索 "ultralytics")
- Bingsu 的 HuggingFace 仓库 用于人脸/人物/服装检测
- Ultralytics 官方资源 用于通用检测模型
- CivitAI 集合(包含 NSFW 内容)
配置设置
路径自定义
在 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" 并选择“安装”
替代方式:手动安装
- 进入目录:
cd custom_nodes
- 克隆仓库:
git clone https://github.com/ltdrdata/ComfyUI-Impact-Subpack
- 进入目录:
cd ComfyUI-Impact-Subpack
- 安装依赖:
pip install -r requirements.txt
- 关键说明:
- 必须使用 ComfyUI 自带的 Python 环境
- 可移植版本需使用
<安装路径>\python_embeded\python.exe -m pip
- 对于虚拟环境,请先激活再使用 pip
- 关键说明:
- 重启 ComfyUI
致谢
核心技术
- ComfyUI - 模块化 Stable Diffusion 接口
模型资源
- Bing-su 的 adetailer - 基于 Ultralytics 的检测模型
- Bingsu 的 HuggingFace 模型 - 其他兼容 Ultralytics 的模型