mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2024-12-03 07:00:47 +08:00
20 lines
494 B
Python
20 lines
494 B
Python
|
from modules import shared
|
||
|
|
|
||
|
|
|
||
|
|
class FaceRestoration:
|
||
|
|
def name(self):
|
||
|
|
return "None"
|
||
|
|
|
||
|
|
def restore(self, np_image):
|
||
|
|
return np_image
|
||
|
|
|
||
|
|
|
||
|
|
def restore_faces(np_image):
|
||
|
|
face_restorers = [x for x in shared.face_restorers if x.name() == shared.opts.face_restoration_model or shared.opts.face_restoration_model is None]
|
||
|
|
if len(face_restorers) == 0:
|
||
|
|
return np_image
|
||
|
|
|
||
|
|
face_restorer = face_restorers[0]
|
||
|
|
|
||
|
|
return face_restorer.restore(np_image)
|