| Tagger for [Automatic1111's WebUI](https://github.com/AUTOMATIC1111/stable-diffusion-webui) | |
| --- | |
| Interrogate booru style tags for single or multiple image files using various models, such as DeepDanbooru. | |
| [한국어를 사용하시나요? 여기에 한국어 설명서가 있습니다!](README.ko.md) | |
| ## Disclaimer | |
| I didn't make any models, and most of the code was heavily borrowed from the [DeepDanbooru](https://github.com/KichangKim/DeepDanbooru) and MrSmillingWolf's tagger. | |
| ## Installation | |
| 1. *Extensions* -> *Install from URL* -> Enter URL of this repository -> Press *Install* button | |
| - or clone this repository under `extensions/` | |
| ```sh | |
| $ git clone https://github.com/toriato/stable-diffusion-webui-wd14-tagger.git extensions/tagger | |
| ``` | |
| 1. *(optional)* Add interrogate model | |
| - #### [*Waifu Diffusion 1.4 Tagger by MrSmilingWolf*](docs/what-is-wd14-tagger.md) | |
| Downloads automatically from the [HuggingFace repository](https://huggingface.co/SmilingWolf/wd-v1-4-vit-tagger) the first time you run it. | |
| - #### *DeepDanbooru* | |
| 1. Various model files can be found below. | |
| - [DeepDanbooru models](https://github.com/KichangKim/DeepDanbooru/releases) | |
| - [e621 model by 🐾Zack🐾#1984](https://discord.gg/BDFpq9Yb7K) | |
| *(link contains NSFW contents!)* | |
| 1. Move the project folder containing the model and config to `models/deepdanbooru` | |
| 1. 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 | |
| │ └╴... | |
| │ | |
| ... | |
| ``` | |
| 1. 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!" | |
| ## Model comparison | |
| [Model comparison](docs/model-comparison.md) | |
| ## Screenshot | |
|  | |
| Artwork made by [hecattaart](https://vk.com/hecattaart?w=wall-89063929_3767) | |
| ## Copyright | |
| Public domain, except borrowed parts (e.g. `dbimutils.py`) | |