64 lines
2.3 KiB
Markdown
64 lines
2.3 KiB
Markdown
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/picobyte/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`)
|