This is a copy of facerestore custom node with a bit of a change to support CodeFormer Fidelity parameter.

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

Git clone this node to ComfyUI/custom_nodes folder:

cd ComfyUI/custom_nodes
git clone https://github.com/mav-rik/facerestore_cf.git

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)

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

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