HF Transformers Classifier Provider

节点功能:该节点它用于从 Hugging Face 加载图像分类模型,并在 ComfyUI 中作为一个标准分类器节点使用。它配合 SEGS_Classify 等节点,可以对图像区域进行标签分类和逻辑判断。HF Transformers Classifier
节点中英文对比
HF Transformers Classifier Provider
  • TRANSFORMERS_CLASSIFIER
  • preset_repo_id
    manual_repo_id
    device_mode
    HF Transformers Classifier Provider
  • TRANSFORMERS_CLASSIFIER
  • 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 这类大型模型上
    暂无节点说明
    节点列表 首页