Upload 5 files

.gitignore

@@ -0,0 +1,6 @@
+__pycache__/
+.vscode/
+.venv/
+.env
+
+presets/

README.ko.md

@@ -0,0 +1,61 @@
+[Automatic1111 웹UI](https://github.com/AUTOMATIC1111/stable-diffusion-webui)를 위한 태깅(라벨링) 확장 기능
+---
+DeepDanbooru 와 같은 모델을 통해 단일 또는 여러 이미지로부터 부루에서 사용하는 태그를 알아냅니다.
+
+[You don't know how to read Korean? Read it in English here!](README.md)
+
+## 들어가기 앞서
+모델과 대부분의 코드는 제가 만들지 않았고 [DeepDanbooru](https://github.com/KichangKim/DeepDanbooru) 와 MrSmillingWolf 의 태거에서 가져왔습니다.
+
+## 설치하기
+1. *확장기능* -> *URL로부터 확장기능 설치* -> 이 레포지토리 주소 입력 -> *설치*
+   - 또는 이 레포지토리를 `extensions/` 디렉터리 내에 클론합니다.
+      ```sh
+      $ git clone https://github.com/toriato/stable-diffusion-webui-wd14-tagger.git extensions/tagger
+      ```
+
+1. 모델 추가하기
+   - #### *MrSmilingWolf's model (a.k.a. Waifu Diffusion 1.4 tagger)*
+      처음 실행할 때 [HuggingFace 레포지토리](https://huggingface.co/SmilingWolf/wd-v1-4-vit-tagger)로부터 자동으로 받아옵니다.
+
+      모델과 관련된 또는 추가 학습에 대한 질문은 원작자인 MrSmilingWolf#5991 으로 물어봐주세요.
+
+   - #### *DeepDanbooru*
+      1. 다양한 모델 파일은 아래 주소에서 찾을 수 있습니다.
+         - [DeepDanbooru model](https://github.com/KichangKim/DeepDanbooru/releases)
+         - [e621 model by 🐾Zack🐾#1984](https://discord.gg/BDFpq9Yb7K)
+            *(NSFW 주의!)*
+
+      1. 모델과 설정 파일이 포함된 프로젝트 폴더를 `models/deepdanbooru` 경로로 옮깁니다.
+
+      1. 파일 구조는 다음과 같습니다:
+         ```
+         models/
+         └╴deepdanbooru/
+           ├╴deepdanbooru-v3-20211112-sgd-e28/
+           │ ├╴project.json
+           │ └╴...
+           │
+           ├╴deepdanbooru-v4-20200814-sgd-e30/
+           │ ├╴project.json
+           │ └╴...
+           │
+           ├╴e621-v3-20221117-sgd-e32/
+           │ ├╴project.json
+           │ └╴...
+           │
+           ...
+         ```
+
+1. 웹UI 를 시작하거나 재시작합니다.
+   - 또는 *Interrogator* 드롭다운 상자 우측에 있는 새로고침 버튼을 누릅니다.
+
+
+## 스크린샷
+![Screenshot](docs/screenshot.png)
+
+Artwork made by [hecattaart](https://vk.com/hecattaart?w=wall-89063929_3767)
+
+## 저작권
+
+빌려온 코드(예: `dbimutils.py`)를 제외하고 모두 Public domain 

README.md

@@ -0,0 +1,63 @@
+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
+![Screenshot](docs/screenshot.png)
+
+Artwork made by [hecattaart](https://vk.com/hecattaart?w=wall-89063929_3767)
+
+## Copyright
+
+Public domain, except borrowed parts (e.g. `dbimutils.py`)

preload.py

@@ -0,0 +1,19 @@
+from pathlib import Path
+from argparse import ArgumentParser
+
+from modules.shared import models_path
+
+default_ddp_path = Path(models_path, 'deepdanbooru')
+
+
+def preload(parser: ArgumentParser):
+    # default deepdanbooru use different paths:
+    # models/deepbooru and models/torch_deepdanbooru
+    # https://github.com/AUTOMATIC1111/stable-diffusion-webui/commit/c81d440d876dfd2ab3560410f37442ef56fc6632
+
+    parser.add_argument(
+        '--deepdanbooru-projects-path',
+        type=str,
+        help='Path to directory with DeepDanbooru project(s).',
+        default=default_ddp_path
+    )

style.css

@@ -0,0 +1,7 @@
+#rating-confidents .output-label>div:not(:first-child) {
+    cursor: pointer;
+}
+
+#tag-confidents .output-label>div:not(:first-child) {
+    cursor: pointer;
+}