stable-diffusion-webui-wd14-tagger

Interrogates single or multiple image files using various alternative models, similar to deepdanbooru interrogate.

Go to file

Roel Kluin baa3c4ecc2 Limit the displayed weighted tags to half the threshold, so we can see some that we're missing, but the list is not overly long		2023-06-05 15:48:59 +02:00
docs	docs: fix typos	2023-01-26 18:36:41 +09:00
javascript	feat: add tag by clicking result label #23	2022-12-11 18:38:54 +09:00
scripts	feat: add basic api #40	2023-01-25 23:47:36 +09:00
tagger	Limit the displayed weighted tags to half the threshold, so we can see	2023-06-05 15:48:59 +02:00
.gitignore	refactor: remove old ignores	2023-01-25 22:22:36 +09:00
README.ko.md	docs: update wd 1.4 model changes	2022-12-06 21:31:23 +09:00
README.md	Update README.md	2023-02-19 08:40:06 +08:00
preload.py	Fix help message	2023-06-04 17:42:19 +02:00
style.css	feat: add tag by clicking result label #23	2022-12-11 18:38:54 +09:00

Tagger for Automatic1111's WebUI

Interrogate booru style tags for single or multiple image files using various models, such as DeepDanbooru.

I didn't make any models, and most of the code was heavily borrowed from the DeepDanbooru and MrSmillingWolf's tagger.

Extensions -> Install from URL -> Enter URL of this repository -> Press Install button
- or clone this repository under extensions/
```
$ git clone https://github.com/toriato/stable-diffusion-webui-wd14-tagger.git extensions/tagger
```
(optional) Add interrogate model
- Waifu Diffusion 1.4 Tagger by MrSmilingWolf
  
  Downloads automatically from the HuggingFace repository the first time you run it.
- DeepDanbooru
  1. Various model files can be found below.
    - DeepDanbooru models
    - e621 model by 🐾Zack🐾#1984 (link contains NSFW contents!)
  2. Move the project folder containing the model and config to models/deepdanbooru
  3. The file structure should look like:
```
models/
└╴deepdanbooru/
  ├╴deepdanbooru-v3-20211112-sgd-e28/
  │ ├╴project.json
  │ └╴...
  │
  ├╴deepdanbooru-v4-20200814-sgd-e30/
  │ ├╴project.json
  │ └╴...
  │
  ├╴e621-v3-20221117-sgd-e32/
  │ ├╴project.json
  │ └╴...
  │
  ...
```
Start or restart the WebUI.
- or you can press refresh button after Interrogator dropdown box.
- "You must close stable diffusion completely after installation and re-run it!"

Artwork made by hecattaart

Public domain, except borrowed parts (e.g. dbimutils.py)