File size: 2,371 Bytes
ca3c8e2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
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`)
|