stable-diffusion-webui-wd14-tagger 是一个为 Automatic1111 Stable Diffusion WebUI 开发的扩展插件,主要用于图像标签识别和分类。
项目地址:https://github.com/toriato/stable-diffusion-webui-wd14-tagger
图像标签识别:能够对单张或多张图像进行批量标签识别
Booru风格标签:生成类似 Danbooru、Gelbooru 等网站的标签格式
模型支持:支持多种预训练模型进行图像识别
1、WD14 Tagger(默认)- 基于 Waifu Diffusion 1.4 的标签器
2、DeepDanbooru(可选)
1、进入 WebUI 的 Extensions 标签页
2、选择 Install from URL
3、输入项目仓库URL
4、点击 Install 按钮
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工作流中特别有用,尤其是在需要大量图像标注和分类的场景下。