第三方插件
ComfyUI-Impact-Pack
HF Transformers Classifier Provider
该节点它用于从 Hugging Face 加载图像分类模型,并在 ComfyUI 中作为一个标准分类器节点使用。它配合 SEGS_Classify 等节点,可以对图像区域进行标签分类和逻辑判断。HF Transformers Classifier
节点中英文对比
按当前节点配置,分别展示中文与英文节点结构。
中文节点
HF Transformers Classifier Provider
preset_repo_id
manual_repo_id
device_mode
英文节点
HF Transformers Classifier Provider
preset_repo_id
manual_repo_id
device_mode
参数说明
依据当前节点关联的 `NodesItems` 数据展示输入、输出与控件说明。
输出参数
TRANSFORMERS_CLASSIFIER
TRANSFORMERS_CLASSIFIER
返回由 HuggingFace pipeline('image-classification') 创建的图像分类模型,可用于后续表达式判定等节点。
控件参数
preset_repo_id
preset_repo_id
预设的 HuggingFace 模型仓库 ID,例如 nateraw/vit-age-classifier、microsoft/resnet-50,也可选 "Manual repo id" 来手动输入。
manual_repo_id
manual_repo_id
如果上方选择了 "Manual repo id",则此字段生效,用于手动填写 HF 仓库路径,如 "hakurei/waifu-classifier"。
device_mode
device_mode
控制推理设备:
- "AUTO"(自动选择 GPU)
- "Prefer GPU"(优先使用 GPU)
- "CPU"(强制使用 CPU)
HF 的 pipeline 会将 device 作为参数传入,如果是 "cpu",速度可能显著下降,特别是在 ViT 或 ResNet 这类大型模型上
详细说明
结合节点用途、参数和调用方式,给出更完整的说明内容。
暂无节点说明
同类节点导航
继续浏览同一分类下的相邻节点,方便串联学习。