2022-09-15 14:53:41 +00:00
|
|
|
---
|
|
|
|
title: Upscale
|
|
|
|
---
|
|
|
|
|
2022-09-18 19:30:18 +00:00
|
|
|
# :material-image-size-select-large: Upscale
|
|
|
|
|
2022-09-15 14:53:41 +00:00
|
|
|
## **GFPGAN and Real-ESRGAN Support**
|
|
|
|
|
|
|
|
The script also provides the ability to do face restoration and upscaling with the help of GFPGAN
|
|
|
|
and Real-ESRGAN respectively.
|
|
|
|
|
|
|
|
As of version 1.14, environment.yaml will install the Real-ESRGAN package into the standard install
|
|
|
|
location for python packages, and will put GFPGAN into a subdirectory of "src" in the
|
|
|
|
stable-diffusion directory. (The reason for this is that the standard GFPGAN distribution has a
|
|
|
|
minor bug that adversely affects image color.) Upscaling with Real-ESRGAN should "just work" without
|
|
|
|
further intervention. Simply pass the --upscale (-U) option on the dream> command line, or indicate
|
|
|
|
the desired scale on the popup in the Web GUI.
|
|
|
|
|
|
|
|
For **GFPGAN** to work, there is one additional step needed. You will need to download and copy the
|
|
|
|
GFPGAN [models file](https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth)
|
|
|
|
into **src/gfpgan/experiments/pretrained_models**. On Mac and Linux systems, here's how you'd do it
|
|
|
|
using **wget**:
|
|
|
|
|
|
|
|
```bash
|
2022-09-11 18:43:07 +00:00
|
|
|
> wget https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.3.pth src/gfpgan/experiments/pretrained_models/
|
2022-09-15 14:53:41 +00:00
|
|
|
```
|
2022-09-11 18:43:07 +00:00
|
|
|
|
|
|
|
Make sure that you're in the stable-diffusion directory when you do this.
|
|
|
|
|
2022-09-15 14:53:41 +00:00
|
|
|
Alternatively, if you have GFPGAN installed elsewhere, or if you are using an earlier version of
|
|
|
|
this package which asked you to install GFPGAN in a sibling directory, you may use the
|
|
|
|
`--gfpgan_dir` argument with `dream.py` to set a custom path to your GFPGAN directory. _There are
|
|
|
|
other GFPGAN related boot arguments if you wish to customize further._
|
2022-09-11 15:52:43 +00:00
|
|
|
|
2022-09-18 02:54:20 +00:00
|
|
|
!!! warning "Internet connection needed"
|
|
|
|
|
|
|
|
Users whose GPU machines are isolated from the Internet (e.g.
|
|
|
|
on a University cluster) should be aware that the first time you run dream.py with GFPGAN and
|
|
|
|
Real-ESRGAN turned on, it will try to download model files from the Internet. To rectify this, you
|
|
|
|
may run `python3 scripts/preload_models.py` after you have installed GFPGAN and all its
|
|
|
|
dependencies.
|
2022-09-11 15:52:43 +00:00
|
|
|
|
2022-09-15 14:53:41 +00:00
|
|
|
## **Usage**
|
2022-09-11 15:52:43 +00:00
|
|
|
|
|
|
|
You will now have access to two new prompt arguments.
|
|
|
|
|
2022-09-15 14:53:41 +00:00
|
|
|
### **Upscaling**
|
2022-09-11 15:52:43 +00:00
|
|
|
|
|
|
|
`-U : <upscaling_factor> <upscaling_strength>`
|
|
|
|
|
2022-09-15 14:53:41 +00:00
|
|
|
The upscaling prompt argument takes two values. The first value is a scaling factor and should be
|
|
|
|
set to either `2` or `4` only. This will either scale the image 2x or 4x respectively using
|
|
|
|
different models.
|
2022-09-11 15:52:43 +00:00
|
|
|
|
2022-09-15 14:53:41 +00:00
|
|
|
You can set the scaling stength between `0` and `1.0` to control intensity of the of the scaling.
|
|
|
|
This is handy because AI upscalers generally tend to smooth out texture details. If you wish to
|
|
|
|
retain some of those for natural looking results, we recommend using values between `0.5 to 0.8`.
|
2022-09-11 15:52:43 +00:00
|
|
|
|
2022-09-15 14:53:41 +00:00
|
|
|
If you do not explicitly specify an upscaling_strength, it will default to 0.75.
|
2022-09-11 15:52:43 +00:00
|
|
|
|
2022-09-15 14:53:41 +00:00
|
|
|
### **Face Restoration**
|
2022-09-11 15:52:43 +00:00
|
|
|
|
|
|
|
`-G : <gfpgan_strength>`
|
|
|
|
|
2022-09-15 14:53:41 +00:00
|
|
|
This prompt argument controls the strength of the face restoration that is being applied. Similar to
|
|
|
|
upscaling, values between `0.5 to 0.8` are recommended.
|
2022-09-11 15:52:43 +00:00
|
|
|
|
2022-09-15 14:53:41 +00:00
|
|
|
You can use either one or both without any conflicts. In cases where you use both, the image will be
|
|
|
|
first upscaled and then the face restoration process will be executed to ensure you get the highest
|
2022-09-11 15:52:43 +00:00
|
|
|
quality facial features.
|
|
|
|
|
|
|
|
`--save_orig`
|
|
|
|
|
2022-09-15 14:53:41 +00:00
|
|
|
When you use either `-U` or `-G`, the final result you get is upscaled or face modified. If you want
|
|
|
|
to save the original Stable Diffusion generation, you can use the `-save_orig` prompt argument to
|
|
|
|
save the original unaffected version too.
|
2022-09-11 15:52:43 +00:00
|
|
|
|
2022-09-15 14:53:41 +00:00
|
|
|
### **Example Usage**
|
2022-09-11 15:52:43 +00:00
|
|
|
|
2022-09-15 14:53:41 +00:00
|
|
|
```bash
|
|
|
|
dream> superman dancing with a panda bear -U 2 0.6 -G 0.4
|
2022-09-11 15:52:43 +00:00
|
|
|
```
|
|
|
|
|
|
|
|
This also works with img2img:
|
|
|
|
|
2022-09-15 14:53:41 +00:00
|
|
|
```bash
|
2022-09-11 15:52:43 +00:00
|
|
|
dream> a man wearing a pineapple hat -I path/to/your/file.png -U 2 0.5 -G 0.6
|
|
|
|
```
|
|
|
|
|
2022-09-18 02:54:20 +00:00
|
|
|
!!! note
|
2022-09-11 15:52:43 +00:00
|
|
|
|
2022-09-18 02:54:20 +00:00
|
|
|
GFPGAN and Real-ESRGAN are both memory intensive. In order to avoid crashes and memory overloads
|
|
|
|
during the Stable Diffusion process, these effects are applied after Stable Diffusion has completed
|
|
|
|
its work.
|
2022-09-11 15:52:43 +00:00
|
|
|
|
2022-09-18 02:54:20 +00:00
|
|
|
In single image generations, you will see the output right away but when you are using multiple
|
|
|
|
iterations, the images will first be generated and then upscaled and face restored after that
|
|
|
|
process is complete. While the image generation is taking place, you will still be able to preview
|
|
|
|
the base images.
|
2022-09-11 15:52:43 +00:00
|
|
|
|
2022-09-15 14:53:41 +00:00
|
|
|
If you wish to stop during the image generation but want to upscale or face restore a particular
|
|
|
|
generated image, pass it again with the same prompt and generated seed along with the `-U` and `-G`
|
|
|
|
prompt arguments to perform those actions.
|
2022-09-18 19:01:05 +00:00
|
|
|
|
|
|
|
## CodeFormer Support
|
|
|
|
|
|
|
|
This repo also allows you to perform face restoration using
|
|
|
|
[CodeFormer](https://github.com/sczhou/CodeFormer).
|
|
|
|
|
|
|
|
In order to setup CodeFormer to work, you need to download the models like with GFPGAN. You can do
|
|
|
|
this either by running `preload_models.py` or by manually downloading the
|
|
|
|
[model file](https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/codeformer.pth) and
|
|
|
|
saving it to `ldm/restoration/codeformer/weights` folder.
|
|
|
|
|
|
|
|
You can use `-ft` prompt argument to swap between CodeFormer and the default GFPGAN. The above
|
|
|
|
mentioned `-G` prompt argument will allow you to control the strength of the restoration effect.
|
|
|
|
|
|
|
|
### **Usage:**
|
|
|
|
|
|
|
|
The following command will perform face restoration with CodeFormer instead of the default gfpgan.
|
|
|
|
|
|
|
|
`<prompt> -G 0.8 -ft codeformer`
|
|
|
|
|
|
|
|
**Other Options:**
|
|
|
|
|
|
|
|
- `-cf` - cf or CodeFormer Fidelity takes values between `0` and `1`. 0 produces high quality
|
|
|
|
results but low accuracy and 1 produces lower quality results but higher accuacy to your original
|
|
|
|
face.
|
|
|
|
|
|
|
|
The following command will perform face restoration with CodeFormer. CodeFormer will output a result
|
|
|
|
that is closely matching to the input face.
|
|
|
|
|
|
|
|
`<prompt> -G 1.0 -ft codeformer -cf 0.9`
|
|
|
|
|
|
|
|
The following command will perform face restoration with CodeFormer. CodeFormer will output a result
|
|
|
|
that is the best restoration possible. This may deviate slightly from the original face. This is an
|
|
|
|
excellent option to use in situations when there is very little facial data to work with.
|
|
|
|
|
|
|
|
`<prompt> -G 1.0 -ft codeformer -cf 0.1`
|