diff --git a/README.md b/README.md index e89e344baf259ae8913fe68ac10abe5eb38d2e94..931bbd71085c295e099bcb82a5d82309ae5a1be7 100644 --- a/README.md +++ b/README.md @@ -40,7 +40,7 @@ We use [EG3D](https://github.com/NVlabs/eg3d) as a pretrained face generator net ```sh $ git clone https://github.com/NVlabs/eg3d.git ``` -We use `ffhq512-128.pkl` [checkpoint](https://catalog.ngc.nvidia.com/orgs/nvidia/teams/research/models/eg3d/files) in our experiments. +We use `ffhqrebalanced512-128.pkl` [checkpoint](https://catalog.ngc.nvidia.com/orgs/nvidia/teams/research/models/eg3d/files) in our experiments. ## Downloading the datasets In our experiments, we use [FFHQ](https://github.com/NVlabs/ffhq-dataset) dataset for training our face reconstruction network. diff --git a/evaluation_pipeline.py b/evaluation_pipeline.py index dc47eadf1d8e5aa6a4d54aee0b2a5d8d5f39b4f9..463d8afe1427f0ccc5812377422eb1ecee7274bb 100644 --- a/evaluation_pipeline.py +++ b/evaluation_pipeline.py @@ -12,8 +12,8 @@ parser.add_argument('--checkpoint', metavar='<checkpoint>', type= str, default=' help='checkpoint of the new mapping network') parser.add_argument('--path_eg3d_repo', metavar='<path_eg3d_repo>', type= str, default='./eg3d', help='./eg3d') -parser.add_argument('--path_eg3d_checkpoint', metavar='<path_eg3d_checkpoint>', type= str, default='./ffhq512-128.pkl', - help='./ffhq512-128.pkl`') +parser.add_argument('--path_eg3d_checkpoint', metavar='<path_eg3d_checkpoint>', type= str, default='./ffhqrebalanced512-128.pkl', + help='./ffhqrebalanced512-128.pkl`') args = parser.parse_args() diff --git a/train.py b/train.py index 40bba9246c76200cf684d9043a50901b67ba32d1..f2eb6f9ef70224d0766de8057c2d62e8582a6bcf 100644 --- a/train.py +++ b/train.py @@ -12,8 +12,8 @@ import argparse parser = argparse.ArgumentParser(description='Train face reconstruction network - GaFaR') parser.add_argument('--path_eg3d_repo', metavar='<path_eg3d_repo>', type= str, default='./eg3d', help='./eg3d') -parser.add_argument('--path_eg3d_checkpoint', metavar='<path_eg3d_checkpoint>', type= str, default='./ffhq512-128.pkl', - help='./ffhq512-128.pkl') +parser.add_argument('--path_eg3d_checkpoint', metavar='<path_eg3d_checkpoint>', type= str, default='./ffhqrebalanced512-128.pkl', + help='./ffhqrebalanced512-128.pkl') parser.add_argument('--path_ffhq_dataset', metavar='<path_ffhq_dataset>', type= str, default='./Flickr-Faces-HQ/images1024x1024', help='FFHQ directory') parser.add_argument('--FR_system', metavar='<FR_system>', type= str, default='ArcFace',