diff options
| author | brkirch <brkirch@users.noreply.github.com> | 2022-10-04 05:04:19 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-10-04 05:04:19 +0000 |
| commit | b88e4ea7d6536ddc814a9101a258f87b06230019 (patch) | |
| tree | 913bdda300b7082aa61429689d0a730cecda1525 /modules/gfpgan_model.py | |
| parent | bdaa36c84470adbdce3e98c01a69af5e95adfb02 (diff) | |
| parent | 2865ef4b9ab16d56326cc805541bebcf01d099bc (diff) | |
| download | stable-diffusion-webui-gfx803-b88e4ea7d6536ddc814a9101a258f87b06230019.tar.gz stable-diffusion-webui-gfx803-b88e4ea7d6536ddc814a9101a258f87b06230019.tar.bz2 stable-diffusion-webui-gfx803-b88e4ea7d6536ddc814a9101a258f87b06230019.zip | |
Merge branch 'master' into master
Diffstat (limited to 'modules/gfpgan_model.py')
| -rw-r--r-- | modules/gfpgan_model.py | 6 |
1 files changed, 1 insertions, 5 deletions
diff --git a/modules/gfpgan_model.py b/modules/gfpgan_model.py index fcd8544a..f1a564b7 100644 --- a/modules/gfpgan_model.py +++ b/modules/gfpgan_model.py @@ -97,11 +97,7 @@ def setup_model(dirname): return "GFPGAN"
def restore(self, np_image):
- np_image_bgr = np_image[:, :, ::-1]
- cropped_faces, restored_faces, gfpgan_output_bgr = gfpgann().enhance(np_image_bgr, has_aligned=False, only_center_face=False, paste_back=True)
- np_image = gfpgan_output_bgr[:, :, ::-1]
-
- return np_image
+ return gfpgan_fix_faces(np_image)
shared.face_restorers.append(FaceRestorerGFPGAN())
except Exception:
|