search

Gourieff / sd-webui-reactor

Fast and Simple Face Swap Extension for StableDiffusion WebUI (A1111 SD WebUI, SD WebUI Forge, SD.Next, Cagliostro)
GNU Affero General Public License v3.0
2.17k stars 235 forks source link

[Feature]: Add GPEN to Face Restoration models #433

Open not-ski opened 3 weeks ago

not-ski commented 3 weeks ago

Feature description

I have tested many face restoration models now (including CodeFormer, GFPGAN, RestoreFormer++, DMDNet, and GPEN) and GPEN has been the best-performing by far. IMO, it outpaces the options currently available (CodeFormer and GFPGAN) by a considerable margin.

GPEN repo can be found here: https://github.com/yangxy/GPEN

It also seems that FaceFusion have made the GPEN models available in ONNX form as well: https://github.com/facefusion/facefusion-assets

The roop-unleashed implementation of GPEN could also potentially be useful: https://github.com/C0untFloyd/roop-unleashed/blob/main/roop/processors/Enhance_GPEN.py

Thanks <3

Gourieff commented 3 weeks ago

Yes, this is in the roadmap But already now you can use ReActor with GPEN via ComfyUI https://github.com/Gourieff/comfyui-reactor-node ReActor Node supports GPEN-BFR-512 ONNX model