Haoming02 / comfyui-old-photo-restoration

Custom Nodes for ComfyUI for Bringing Old Photo Back to Life
MIT License
14 stars 4 forks source link
comfyui custom-nodes stable-diffusion

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


example workflow
(You can drag this image directly into ComfyUI)

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 the packages in requirement.txt
  2. Download global_checkpoints.zip from Releases
  3. Extract and put the checkpoints folder (not just the files) into ~ComfyUI/custom_nodes/comfyui-old-photo-restoration/Global
  4. Download face_checkpoints.zip from Releases
  5. Extract and put the checkpoints folder (not just the files) into ~ComfyUI/custom_nodes/comfyui-old-photo-restoration/Face_Enhancement
  6. Download shape_predictor_68_face_landmarks.zip from Releases
  7. Extract the .dat file into ~ComfyUI/custom_nodes/comfyui-old-photo-restoration/Face_Detection

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

Another mirror: Google Drive

How to Use

Simply connect an image to the Global Restoration node to process; if the image contains scratch artifacts, use the Global Restoration with Scratch Processing node instead.

To improve the faces, connect the image to the Face Detection node then the Face Enhancement node, then connect the enhanced output and the image both to the Face Align node in the end to merge the results.

Settings