From c7ccc098a7e3120eb666050e991ae111dcdadf34 Mon Sep 17 00:00:00 2001 From: Xintao Date: Tue, 7 Jun 2022 16:49:26 +0800 Subject: [PATCH] update facelib; use seg to paste back --- gfpgan/utils.py | 1 + requirements.txt | 14 +++++++------- 2 files changed, 8 insertions(+), 7 deletions(-) diff --git a/gfpgan/utils.py b/gfpgan/utils.py index 7a2c1c9..35658c5 100644 --- a/gfpgan/utils.py +++ b/gfpgan/utils.py @@ -79,6 +79,7 @@ class GFPGANer(): crop_ratio=(1, 1), det_model='retinaface_resnet50', save_ext='png', + use_parse=True, device=self.device) if model_path.startswith('https://'): diff --git a/requirements.txt b/requirements.txt index e6861b8..3edf9e0 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,12 +1,12 @@ -torch>=1.7 +basicsr>=1.3.4.0 +facexlib>=0.2.3 +lmdb numpy<1.21 # numba requires numpy<1.21,>=1.17 opencv-python -torchvision -scipy -tqdm -basicsr>=1.3.4.0 -facexlib>=0.2.0.3 -lmdb pyyaml +scipy tb-nightly +torch>=1.7 +torchvision +tqdm yapf