dikdimon's picture
Upload extensions using SD-Hub extension
f4a41d8 verified
|
raw
history blame
4.24 kB
# ๋š๊ฑฐ Detection Detailer
Dotgeo(hijack) Detection Detailer
ddetailer with torch 2.0, mmcv 2.0, mmdet 3.0
integrated with [noahge4/ddetailer](https://github.com/noahge4/ddetailer)
AI์‹ค์‚ฌ์ฑ„๋„ ChatGPT23๋‹˜์˜ [ddetailer ์ˆ˜์ •๋ณธ](https://arca.live/b/aireal/72297207) ๋ณ‘ํ•ฉ๋จ
## Installation
1. remove original ddetailer extension - `stable-diffusion-webui/extensions/ddetailer` folder
2. remove original model files - `stable-diffusion-webui/models/mmdet` folder
3. install from the extensions tab with url `https://github.com/Bing-su/dddetailer`
## Problem
The predictive accuracy of the segmentation model has become very poor.
# Detection Detailer
An object detection and auto-mask extension for [Stable Diffusion web UI](https://github.com/AUTOMATIC1111/stable-diffusion-webui). See [Installation](https://github.com/dustysys/ddetailer#installation).
![adoringfan](/misc/ddetailer_example_1.png)
### Segmentation
Default models enable person and face instance segmentation.
![amgothic](/misc/ddetailer_example_2.png)
### Detailing
With full-resolution inpainting, the extension is handy for improving faces without the hassle of manual masking.
![zion](/misc/ddetailer_example_3.gif)
## Installation
1. Use `git clone https://github.com/dustysys/ddetailer.git` from your SD web UI `/extensions` folder. Alternatively, install from the extensions tab with url `https://github.com/dustysys/ddetailer`
2. Start or reload SD web UI.
The models and dependencies should download automatically. To install them manually, follow the [official instructions for installing mmdet](https://mmcv.readthedocs.io/en/latest/get_started/installation.html#install-with-mim-recommended). The models can be [downloaded here](https://huggingface.co./dustysys/ddetailer) and should be placed in `/models/mmdet/bbox` for bounding box (`anime-face_yolov3`) or `/models/mmdet/segm` for instance segmentation models (`dd-person_mask2former`). See the [MMDetection docs](https://mmdetection.readthedocs.io/en/latest/1_exist_data_model.html) for guidance on training your own models. For using official MMDetection pretrained models see [here](https://github.com/dustysys/ddetailer/issues/5#issuecomment-1311231989), these are trained for photorealism. See [Troubleshooting](https://github.com/dustysys/ddetailer#troubleshooting) if you encounter issues during installation.
## Usage
Select Detection Detailer as the script in SD web UI to use the extension. Click 'Generate' to run the script. Here are some tips:
- `anime-face_yolov3` can detect the bounding box of faces as the primary model while `dd-person_mask2former` isolates the head's silhouette as the secondary model by using the bitwise AND option. Refer to [this example](https://github.com/dustysys/ddetailer/issues/4#issuecomment-1311200268).
- The dilation factor expands the mask, while the x & y offsets move the mask around.
- The script is available in txt2img mode as well and can improve the quality of your 10 pulls with moderate settings (low denoise).
## Troubleshooting
If you get the message ERROR: 'Failed building wheel for pycocotools' follow [these steps](https://github.com/dustysys/ddetailer/issues/1#issuecomment-1309415543).
Any other issues installing, open an [issue](https://github.com/dustysys/ddetailer/issues).
## Credits
hysts/[anime-face-detector](https://github.com/hysts/anime-face-detector) - Creator of `anime-face_yolov3`, which has impressive performance on a variety of art styles.
skytnt/[anime-segmentation](https://huggingface.co./datasets/skytnt/anime-segmentation) - Synthetic dataset used to train `dd-person_mask2former`.
jerryli27/[AniSeg](https://github.com/jerryli27/AniSeg) - Annotated dataset used to train `dd-person_mask2former`.
open-mmlab/[mmdetection](https://github.com/open-mmlab/mmdetection) - Object detection toolset. `dd-person_mask2former` was trained via transfer learning using their [R-50 Mask2Former instance segmentation model](https://github.com/open-mmlab/mmdetection/tree/master/configs/mask2former#instance-segmentation) as a base.
AUTOMATIC1111/[stable-diffusion-webui](https://github.com/AUTOMATIC1111/stable-diffusion-webui) - Web UI for Stable Diffusion, base application for this extension.