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