Go to https://github.com/mav-rik/facerestore_cf for a more up to date fork of this project.

These ComfyUI nodes can be used to restore faces in images similar to the face restore option in AUTOMATIC1111 webui.

Installation

Extract the zip and put the facerestore directory inside the ComfyUI custom_nodes directory.

If running the portable windows version of ComfyUI, run embedded_install.bat

Windows users not running the portable version can run install.bat. Linux and mac users can run install.sh

If the install fails for whatever reason, you'll need to work out how to install opencv-python yourself (you'll get a cv2 not found error otherwise)

Face Restoration Models

These models perform the face restoration. If you use automatic1111 you probably already have them but if not here are the download links.

Just put them in models/facerestore_models.

https://github.com/TencentARC/GFPGAN/releases/

https://github.com/TencentARC/GFPGAN/releases/download/v1.3.4/GFPGANv1.4.pth

https://github.com/sczhou/CodeFormer/releases/

https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/codeformer.pth

Face detection models

The face restoration model only works with cropped face images. A face detection model is used to send a crop of each face found to the face restoration model.

These will automaticly be downloaded and placed in models/facedetection the first time each is used.

Here's the links if you'd rather download them yourself.

https://github.com/xinntao/facexlib/releases/download/v0.1.0/detection_Resnet50_Final.pth

https://github.com/xinntao/facexlib/releases/download/v0.1.0/detection_mobilenet0.25_Final.pth

https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/yolov5l-face.pth

https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/yolov5n-face.pth

This also gets downloaded and put in the facedetection folder. It's used to put the restored faces back into the original image.

https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/parsing_parsenet.pth