SD图像标签神器:stable-diffusion-webui-wd14-tagger插件详解与安装指南

2024-11-23 人工智能 485 次阅读 0 次点赞
stable-diffusion-webui-wd14-tagger是为Stable Diffusion WebUI开发的扩展插件,用于图像标签识别和分类。它支持单张或批量图像处理,生成Booru风格标签,并支持多种预训练模型,包括默认的WD14 Tagger(基于Waifu Diffusion 1.4)和可选的DeepDanbooru。用户可通过WebUI扩展安装或Git克隆方式部署。原项目已停止维护,建议寻找其他维护分支。该工具特别适用于训练LoRA等模型的标签数据准备、图像自动分类及工作流优化。需注意部分模型可能包含NSFW内容,安装后需完全重启WebUI。

stable-diffusion-webui-wd14-tagger 是一个为 Automatic1111 Stable Diffusion WebUI 开发的扩展插件,主要用于图像标签识别和分类。

项目地址

项目地址:https://github.com/toriato/stable-diffusion-webui-wd14-tagger

核心功能

图像标签识别:能够对单张或多张图像进行批量标签识别

Booru风格标签:生成类似 Danbooru、Gelbooru 等网站的标签格式

模型支持:支持多种预训练模型进行图像识别

支持的模型

WD14 Tagger(默认)- 基于 Waifu Diffusion 1.4 的标签器

首次使用时自动从 HuggingFace 下载,支持广泛的动漫风格图像标签。

DeepDanbooru(可选)

需要手动下载模型文件,支持多种变体(v3、v4等),还包括 e621 模型(包含NSFW内容)。

安装方法

方式一:通过WebUI扩展安装

1、进入 WebUI 的 Extensions 标签页

2、选择 Install from URL

3、输入项目仓库URL

4、点击 Install 按钮

方式二:Git克隆

git clone https://github.com/toriato/stable-diffusion-webui-wd14-tagger.git extensions/tagger

项目状态

重要说明:根据提供的参考资料,原项目(toriato版本)已经停止维护。用户可能需要考虑:

  • 使用其他维护中的分支版本
  • 寻找替代的标签识别扩展
  • 关注项目的新维护者(如picobyte的分支)

使用场景

这个扩展特别适用于:

训练数据准备:为训练LoRA、Textual Inversion等模型准备标签数据。

图像分类管理:对生成的图像进行自动分类和标记。

工作流优化:批量处理大量图像的标签识别任务。

技术特点

自动化下载:WD14模型自动从HuggingFace下载。

多模型支持:可配置不同的识别模型。

批量处理:支持单张或批量图像处理。

集成界面:直接在WebUI中操作,无需额外工具。

注意事项

  • 项目基于现有的DeepDanbooru等开源代码构建。
  • 安装后需要完全重启WebUI才能生效。
  • 部分模型可能包含NSFW内容,使用时请注意。
  • 这个扩展在Stable Diffusion工作流中特别有用,尤其是在需要大量图像标注和分类的场景下。
最后更新于23天前
本文由人工编写,AI优化,转载请注明原文地址: SD图像标签神器:stable-diffusion-webui-wd14-tagger插件详解与安装指南

评论 (2)

登录 后发表评论

书虫先生书虫先生2025-12-04 09:45:09

教程很详细,特别是关于不同模型和安装方式的说明,对我整理图库帮助很大。感谢作者分享!不过原项目停止维护了,后续使用有替代方案推荐吗?

张小凡张小凡2025-11-24 11:31:07

这个扩展太实用了!自动打标签功能为我的LoRA训练节省了大量时间。感谢作者的分享和教程,让我这种新手也能轻松上手。想问下现在项目停止维护了,有没有推荐的替代分支呢?