nod-ai / SHARK

SHARK - High Performance Machine Learning Distribution
Apache License 2.0
1.41k stars 168 forks source link

Add face restore / colorize #791

Open rexorp opened 1 year ago

rexorp commented 1 year ago

I found these options to be very useful for my creative applications when restoring old photos.

powderluv commented 1 year ago

Image to image is done.

powderluv commented 1 year ago

Upscaling is also in. (modulo some bugs we can open new issues for). Will rename this to face restore

cstueckrath commented 1 year ago

while it is nice to have the upscaler as separate tab, it should eventually be merged into something like post-processing and hold all the options for that. Also, regarding face restoration: this needs to have a checkbox in at least txt2img so it can be run automatically after each generation.

powderluv commented 1 year ago

ah good point on the checkbox. What are the latest models for Face restoration ?

cstueckrath commented 1 year ago

GFPGAN 1.3/1.4 and CodeFormer is still SOTA afaik. I'm not that into that topic, though. But I know those work well.

powderluv commented 1 year ago

codeformer

cstueckrath commented 1 year ago

gfpgan

cstueckrath commented 1 year ago

IMO they work best when used together. I usually use only gfpgan, but give it another pass in codeformer when the output is too model like

cstueckrath commented 1 year ago

found another interesting one: DifFace

RousseauRemi commented 1 year ago

When do the face restoration will be availlable ?