diff --git a/gfpgan/utils.py b/gfpgan/utils.py index 7a2c1c9a..35658c52 100644 --- a/gfpgan/utils.py +++ b/gfpgan/utils.py @@ -79,6 +79,7 @@ def __init__(self, model_path, upscale=2, arch='clean', channel_multiplier=2, bg 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 e6861b8c..3edf9e03 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,12 +1,12 @@ -torch>=1.7 -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 +facexlib>=0.2.3 lmdb +numpy<1.21 # numba requires numpy<1.21,>=1.17 +opencv-python pyyaml +scipy tb-nightly +torch>=1.7 +torchvision +tqdm yapf