舒适UI WD 1.4 标签器

一个基于 ComfyUI 的扩展，允许从图片中查询 booru 标签。

基于 SmilingWolf/wd-v1-4-tags 和 toriato/stable-diffusion-webui-wd14-tagger
所有模型由 SmilingWolf 创建

安装

1. 将 git clone https://github.com/pythongosssss/ComfyUI-WD14-Tagger 克隆到 custom_nodes 文件夹中
   • 例如 custom_nodes\ComfyUI-WD14-Tagger
2. 打开命令提示符/终端等
3. 切换到你刚刚创建的 custom_nodes\ComfyUI-WD14-Tagger 文件夹
   • 例如 cd C:\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI-WD14-Tagger 或者你安装的任何位置
4. 安装 Python 包
   • Windows 独立安装（嵌入式 Python）:
     ../../../python_embeded/python.exe -s -m pip install -r requirements.txt
   • 手动/非 Windows 安装
     pip install -r requirements.txt

使用

通过 image -> WD14Tagger|pysssss 添加节点

如果模型缺失，运行时将自动下载。

支持为多个批量输入进行标签和输出。

• model：要使用的查询模型。你可以在这里尝试它们 WaifuDiffusion v1.4 Tags。最新模型（截至撰写时）是 MOAT，最受欢迎的是 ConvNextV2。
• threshold：标签被视为有效的分数
• character_threshold：角色标签被视为有效的分数
• exclude_tags：应以逗号分隔的，不应包含在结果中的标签列表

任何显示图片的节点（例如 LoadImage、SaveImage、PreviewImage）也可以快速查询图片。
只需右击节点（如果是显示多个图片，则单击要查询的图片），然后从菜单中选择 WD14 Tagger

这些设置在 pysssss.json 文件的 settings 部分中。

离线使用

最简单的方法是在线使用，查询一张图片，模型将下载并缓存，但如果您想手动下载模型：

• 创建一个 models 文件夹（与 wd14tagger.py 在同一目录下）
• 使用 pysssss.json 列表中的模型 URL
• 下载 model.onnx 并以模型名称命名，例如 wd-v1-4-convnext-tagger-v2.onnx
• 下载 selected_tags.csv 并以模型名称命名，例如 wd-v1-4-convnext-tagger-v2.csv

要求

onnxruntime（推荐，即使是在 CPU 上查询速度也很快，已包含在 requirements.txt 中）
或者 onnxruntime-gpu（允许使用 GPU，许多人使用时出现问题，如果你尝试，我无法提供支持）

更新日志

• 2023-05-14 - 移动到自己的仓库，添加下载模型，支持多个输入