search

AUTOMATIC1111 / stable-diffusion-webui

Stable Diffusion web UI
GNU Affero General Public License v3.0
131.71k stars 25.28k forks source link

[Feature Request]: Add GPEN to Face Restoration models #15630

Open not-ski opened 3 weeks ago

not-ski commented 3 weeks ago

Is there an existing issue for this?

What would your feature do ?

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 in Automatic1111 (CodeFormer and GFPGAN) by a considerable degree.

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

Proposed workflow

  1. Go to ....
  2. Press ....
  3. ???
  4. Profit

Additional information

No response

ThereforeGames commented 3 weeks ago

I agree. GPEN is superior to other face restoration techniques in most cases. I implemented it in my own extension, Unprompted, but it would be great to have as a native option.

Docs on using it through the extension, in case anyone's interested: https://github.com/ThereforeGames/unprompted/wiki/Shortcodes#restore_faces

There is also a GUI version for the code-averse.