Haoming02 / sd-webui-old-photo-restoration

An Extension for Automatic1111 Webui for Bringing Old Photo Back to Life
MIT License
86 stars 5 forks source link
stable-diffusion-webui stable-diffusion-webui-plugin

# SD Webui Old Photo Restoration This is an Extension for the Automatic1111 Webui, which allows you to perform Bringing-Old-Photos-Back-to-Life natively.

Compatible with Forge

Original Paper: https://arxiv.org/abs/2004.09484

Original Repo: https://github.com/microsoft/Bringing-Old-Photos-Back-to-Life (includes some example images)

Requirements

  1. Install this Extension*
  2. Download global_checkpoints.zip from Releases
  3. Extract and put the checkpoints folder (not just the files) into ~webui/extensions/sd-webui-old-photo-restoration/Global
  4. Download face_checkpoints.zip from Releases
  5. Extract and put the checkpoints folder (not just the files) into ~webui/extensions/sd-webui-old-photo-restoration/Face_Enhancement
  6. Download shape_predictor_68_face_landmarks.zip from Releases
  7. Extract the .dat file into ~webui/extensions/sd-webui-old-photo-restoration/Face_Detection

The Releases page includes the original links, as well as the backups mirrored by myself

Another mirror: Google Drive

Extension not Showing Up

If after installing the Extension, nothing shows up in the Extras tab, it's usually caused by failing to install the dlib package. To solve it:

Linux

From my brief searching, you need to install cmake first, then try installing dlib again

pip install dlib

or

sudo apt-get install cmake

Windows

Easier Way

Intended Way

How to Use

After installing this Extension, there will be an Old Photo Restoration section in the Extras tab

  1. Expand the dropdown to enable the features
  2. Upload the image(s) to be processed
  3. Adjust the settings as needed
  4. Click Generate
  5. The result(s) will show up on the right once the process finishes

Settings