Tagger for Automatic1111's WebUI

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

한국어를 사용하시나요? 여기에 한국어 설명서가 있습니다!

Disclaimer

I didn't make any models, and most of the code was heavily borrowed from the 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/

     $ git clone https://github.com/toriato/stable-diffusion-webui-wd14-tagger.git extensions/tagger
     

2. Add interrogate model

   • 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
          │ └╴...
          │
          ...
        

   • MrSmilingWolf's model (a.k.a. Waifu Diffusion 1.4 tagger)

     Please ask the original author MrSmilingWolf#5991 for questions related to model or additional training.

3. Start or restart the WebUI.

   • or you can press refresh button after Interrogator dropdown box.

Screenshot

Artwork made by hecattaart

Copyright

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