ComfyUI-WD14-Tagger 安装和配置指南

高保其Kimball

于 2024-09-13 22:13:49 发布

阅读量272

点赞数 4

本文链接：https://blog.csdn.net/gitblog_09201/article/details/142229486

版权

ComfyUI-WD14-Tagger 安装和配置指南

ComfyUI-WD14-Tagger A ComfyUI extension allowing for the interrogation of booru tags from images. 项目地址: https://gitcode.com/gh_mirrors/co/ComfyUI-WD14-Tagger

1. 项目基础介绍和主要编程语言

项目基础介绍

ComfyUI-WD14-Tagger 是一个基于 ComfyUI 的扩展插件，旨在从图像中提取 booru 标签。该项目基于 SmilingWolf/wd-v1-4-tags 和 toriato/stable-diffusion-webui-wd14-tagger 开发，能够自动下载和使用预训练模型进行图像标签的提取。

主要编程语言

该项目主要使用 Python 语言进行开发。

2. 项目使用的关键技术和框架

关键技术

ComfyUI: 一个用于图像处理的 UI 框架。
ONNX Runtime: 用于模型推理的高性能引擎。
Python: 主要编程语言，用于实现图像标签提取的逻辑。

框架

SmilingWolf/wd-v1-4-tags: 提供基础的图像标签提取模型。
toriato/stable-diffusion-webui-wd14-tagger: 提供图像标签提取的 WebUI 实现。

3. 项目安装和配置的准备工作和详细安装步骤

准备工作

安装 Python: 确保你的系统中已经安装了 Python 3.7 或更高版本。
安装 Git: 用于克隆项目代码。
安装依赖包: 项目依赖于一些 Python 包，需要通过 pip 安装。

详细安装步骤

步骤 1: 克隆项目代码

首先，打开命令行工具（如 Windows 的 Command Prompt 或 PowerShell，Linux/Mac 的 Terminal），然后运行以下命令克隆项目代码：

git clone https://github.com/pythongosssss/ComfyUI-WD14-Tagger.git

步骤 2: 进入项目目录

克隆完成后，进入项目目录：

cd ComfyUI-WD14-Tagger

步骤 3: 安装 Python 依赖包

根据你的系统环境，选择合适的安装方式：

Windows Standalone 安装（嵌入式 Python）: 如果你的系统中使用了嵌入式 Python，可以使用以下命令安装依赖包：
```
.\python_embeded\python.exe -s -m pip install -r requirements.txt
```
手动/非 Windows 安装: 如果你使用的是非 Windows 系统，或者没有嵌入式 Python，可以使用以下命令安装依赖包：
```
pip install -r requirements.txt
```

步骤 4: 配置和使用

安装完成后，你可以通过以下步骤配置和使用 ComfyUI-WD14-Tagger：

添加节点: 在 ComfyUI 中，通过 image -> WD14Tagger|pysssss 添加节点。
模型下载: 模型会在运行时自动下载，如果需要手动下载，可以参考项目文档中的说明。
设置参数: 你可以设置 threshold、character_threshold 和 exclude_tags 等参数来调整标签提取的行为。