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
HF Transformers Classifier Provider - 参数说明
输出参数
TRANSFORMERS_CLASSIFIER
返回由 HuggingFace pipeline('image-classification') 创建的图像分类模型,可用于后续表达式判定等节点。
控件参数
preset_repo_id
预设的 HuggingFace 模型仓库 ID,例如 nateraw/vit-age-classifier、microsoft/resnet-50,也可选 "Manual repo id" 来手动输入。
manual_repo_id
如果上方选择了 "Manual repo id",则此字段生效,用于手动填写 HF 仓库路径,如 "hakurei/waifu-classifier"。
device_mode
控制推理设备:
- "AUTO"(自动选择 GPU)
- "Prefer GPU"(优先使用 GPU)
- "CPU"(强制使用 CPU)
HF 的 pipeline 会将 device 作为参数传入,如果是 "cpu",速度可能显著下降,特别是在 ViT 或 ResNet 这类大型模型上
暂无节点说明