diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-09-07 09:32:28 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-09-07 09:32:28 +0000 |
| commit | 6a9b33c848281cb02f38764e4f91ef767f5e3edd (patch) | |
| tree | d3af251b9e38b0187f2cfe6ad4d81bb5e3ae3eeb /modules/face_restoration.py | |
| parent | 9cb3cc3a2f5f419dd594f3322fa35113a6ed2391 (diff) | |
| download | stable-diffusion-webui-gfx803-6a9b33c848281cb02f38764e4f91ef767f5e3edd.tar.gz stable-diffusion-webui-gfx803-6a9b33c848281cb02f38764e4f91ef767f5e3edd.tar.bz2 stable-diffusion-webui-gfx803-6a9b33c848281cb02f38764e4f91ef767f5e3edd.zip | |
codeformer support
Diffstat (limited to 'modules/face_restoration.py')
| -rw-r--r-- | modules/face_restoration.py | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/modules/face_restoration.py b/modules/face_restoration.py new file mode 100644 index 00000000..4ae53d21 --- /dev/null +++ b/modules/face_restoration.py @@ -0,0 +1,19 @@ +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)
|