├── .gitignore ├── ACKNOWLEDGEMENTS ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── configs └── gmpi.yml ├── data_preprocess ├── prepare_landmarks_ffhq.py └── prepare_landmarks_metfaces.py ├── docs └── TRAIN_EVAL.md ├── environment.yml ├── gmpi ├── core │ ├── __init__.py │ ├── camera.py │ ├── light_renderer.py │ ├── mpi.py │ └── mpi_renderer.py ├── curriculums.py ├── datasets.py ├── eval │ ├── common.py │ ├── compute_consistency.py │ ├── compute_fid_kid.py │ ├── compute_geometry.py │ ├── deep3dface_gmpi.py │ ├── eval.sh │ ├── extract_mesh.py │ ├── prepare_face_landmarks.py │ ├── prepare_fake_data.py │ ├── prepare_real_data.py │ └── vis │ │ ├── extract_mesh.py │ │ └── render_video.py ├── fid_evaluation.py ├── models │ ├── dnnlib │ │ ├── __init__.py │ │ └── util.py │ ├── legacy.py │ ├── networks │ │ ├── __init__.py │ │ ├── networks_cond_on_pos_enc.py │ │ ├── networks_pos_enc_learnable_param.py │ │ ├── networks_vanilla.py │ │ └── networks_vanilla_depth2alpha.py │ └── torch_utils │ │ ├── __init__.py │ │ ├── custom_ops.py │ │ ├── misc.py │ │ ├── ops │ │ ├── __init__.py │ │ ├── bias_act.cpp │ │ ├── bias_act.cu │ │ ├── bias_act.h │ │ ├── bias_act.py │ │ ├── conv2d_gradfix.py │ │ ├── conv2d_resample.py │ │ ├── fma.py │ │ ├── grid_sample_gradfix.py │ │ ├── upfirdn2d.cpp │ │ ├── upfirdn2d.cu │ │ ├── upfirdn2d.h │ │ └── upfirdn2d.py │ │ └── persistence.py ├── train.py ├── train_helpers.py └── utils │ ├── __init__.py │ ├── cam_utils.py │ ├── color_grad.py │ ├── config.py │ ├── img_utils.py │ ├── io_utils.py │ ├── logging.py │ ├── mpi_utils.py │ ├── pytorch_fid │ ├── __init__.py │ ├── fid_score.py │ └── inception.py │ ├── registry.py │ ├── tensorboard_utils.py │ └── torch_utils.py ├── launch.py ├── media └── teaser.jpg ├── run_gmpi.py ├── setup.py └── virtual_envs ├── deep3dface_env.yaml ├── deepface_env.yaml └── mtcnn_env.yaml /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/.gitignore -------------------------------------------------------------------------------- /ACKNOWLEDGEMENTS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/ACKNOWLEDGEMENTS -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/README.md -------------------------------------------------------------------------------- /configs/gmpi.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/configs/gmpi.yml -------------------------------------------------------------------------------- /data_preprocess/prepare_landmarks_ffhq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/data_preprocess/prepare_landmarks_ffhq.py -------------------------------------------------------------------------------- /data_preprocess/prepare_landmarks_metfaces.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/data_preprocess/prepare_landmarks_metfaces.py -------------------------------------------------------------------------------- /docs/TRAIN_EVAL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/docs/TRAIN_EVAL.md -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/environment.yml -------------------------------------------------------------------------------- /gmpi/core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gmpi/core/camera.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/core/camera.py -------------------------------------------------------------------------------- /gmpi/core/light_renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/core/light_renderer.py -------------------------------------------------------------------------------- /gmpi/core/mpi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/core/mpi.py -------------------------------------------------------------------------------- /gmpi/core/mpi_renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/core/mpi_renderer.py -------------------------------------------------------------------------------- /gmpi/curriculums.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/curriculums.py -------------------------------------------------------------------------------- /gmpi/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/datasets.py -------------------------------------------------------------------------------- /gmpi/eval/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/eval/common.py -------------------------------------------------------------------------------- /gmpi/eval/compute_consistency.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/eval/compute_consistency.py -------------------------------------------------------------------------------- /gmpi/eval/compute_fid_kid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/eval/compute_fid_kid.py -------------------------------------------------------------------------------- /gmpi/eval/compute_geometry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/eval/compute_geometry.py -------------------------------------------------------------------------------- /gmpi/eval/deep3dface_gmpi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/eval/deep3dface_gmpi.py -------------------------------------------------------------------------------- /gmpi/eval/eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/eval/eval.sh -------------------------------------------------------------------------------- /gmpi/eval/extract_mesh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/eval/extract_mesh.py -------------------------------------------------------------------------------- /gmpi/eval/prepare_face_landmarks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/eval/prepare_face_landmarks.py -------------------------------------------------------------------------------- /gmpi/eval/prepare_fake_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/eval/prepare_fake_data.py -------------------------------------------------------------------------------- /gmpi/eval/prepare_real_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/eval/prepare_real_data.py -------------------------------------------------------------------------------- /gmpi/eval/vis/extract_mesh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/eval/vis/extract_mesh.py -------------------------------------------------------------------------------- /gmpi/eval/vis/render_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/eval/vis/render_video.py -------------------------------------------------------------------------------- /gmpi/fid_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/fid_evaluation.py -------------------------------------------------------------------------------- /gmpi/models/dnnlib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/models/dnnlib/__init__.py -------------------------------------------------------------------------------- /gmpi/models/dnnlib/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/models/dnnlib/util.py -------------------------------------------------------------------------------- /gmpi/models/legacy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/models/legacy.py -------------------------------------------------------------------------------- /gmpi/models/networks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/models/networks/__init__.py -------------------------------------------------------------------------------- /gmpi/models/networks/networks_cond_on_pos_enc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/models/networks/networks_cond_on_pos_enc.py -------------------------------------------------------------------------------- /gmpi/models/networks/networks_pos_enc_learnable_param.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/models/networks/networks_pos_enc_learnable_param.py -------------------------------------------------------------------------------- /gmpi/models/networks/networks_vanilla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/models/networks/networks_vanilla.py -------------------------------------------------------------------------------- /gmpi/models/networks/networks_vanilla_depth2alpha.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/models/networks/networks_vanilla_depth2alpha.py -------------------------------------------------------------------------------- /gmpi/models/torch_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/models/torch_utils/__init__.py -------------------------------------------------------------------------------- /gmpi/models/torch_utils/custom_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/models/torch_utils/custom_ops.py -------------------------------------------------------------------------------- /gmpi/models/torch_utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/models/torch_utils/misc.py -------------------------------------------------------------------------------- /gmpi/models/torch_utils/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/models/torch_utils/ops/__init__.py -------------------------------------------------------------------------------- /gmpi/models/torch_utils/ops/bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/models/torch_utils/ops/bias_act.cpp -------------------------------------------------------------------------------- /gmpi/models/torch_utils/ops/bias_act.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/models/torch_utils/ops/bias_act.cu -------------------------------------------------------------------------------- /gmpi/models/torch_utils/ops/bias_act.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/models/torch_utils/ops/bias_act.h -------------------------------------------------------------------------------- /gmpi/models/torch_utils/ops/bias_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/models/torch_utils/ops/bias_act.py -------------------------------------------------------------------------------- /gmpi/models/torch_utils/ops/conv2d_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/models/torch_utils/ops/conv2d_gradfix.py -------------------------------------------------------------------------------- /gmpi/models/torch_utils/ops/conv2d_resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/models/torch_utils/ops/conv2d_resample.py -------------------------------------------------------------------------------- /gmpi/models/torch_utils/ops/fma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/models/torch_utils/ops/fma.py -------------------------------------------------------------------------------- /gmpi/models/torch_utils/ops/grid_sample_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/models/torch_utils/ops/grid_sample_gradfix.py -------------------------------------------------------------------------------- /gmpi/models/torch_utils/ops/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/models/torch_utils/ops/upfirdn2d.cpp -------------------------------------------------------------------------------- /gmpi/models/torch_utils/ops/upfirdn2d.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/models/torch_utils/ops/upfirdn2d.cu -------------------------------------------------------------------------------- /gmpi/models/torch_utils/ops/upfirdn2d.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/models/torch_utils/ops/upfirdn2d.h -------------------------------------------------------------------------------- /gmpi/models/torch_utils/ops/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/models/torch_utils/ops/upfirdn2d.py -------------------------------------------------------------------------------- /gmpi/models/torch_utils/persistence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/models/torch_utils/persistence.py -------------------------------------------------------------------------------- /gmpi/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/train.py -------------------------------------------------------------------------------- /gmpi/train_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/train_helpers.py -------------------------------------------------------------------------------- /gmpi/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/utils/__init__.py -------------------------------------------------------------------------------- /gmpi/utils/cam_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/utils/cam_utils.py -------------------------------------------------------------------------------- /gmpi/utils/color_grad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/utils/color_grad.py -------------------------------------------------------------------------------- /gmpi/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/utils/config.py -------------------------------------------------------------------------------- /gmpi/utils/img_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/utils/img_utils.py -------------------------------------------------------------------------------- /gmpi/utils/io_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/utils/io_utils.py -------------------------------------------------------------------------------- /gmpi/utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/utils/logging.py -------------------------------------------------------------------------------- /gmpi/utils/mpi_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/utils/mpi_utils.py -------------------------------------------------------------------------------- /gmpi/utils/pytorch_fid/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /gmpi/utils/pytorch_fid/fid_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/utils/pytorch_fid/fid_score.py -------------------------------------------------------------------------------- /gmpi/utils/pytorch_fid/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/utils/pytorch_fid/inception.py -------------------------------------------------------------------------------- /gmpi/utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/utils/registry.py -------------------------------------------------------------------------------- /gmpi/utils/tensorboard_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/utils/tensorboard_utils.py -------------------------------------------------------------------------------- /gmpi/utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/gmpi/utils/torch_utils.py -------------------------------------------------------------------------------- /launch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/launch.py -------------------------------------------------------------------------------- /media/teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/media/teaser.jpg -------------------------------------------------------------------------------- /run_gmpi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/run_gmpi.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/setup.py -------------------------------------------------------------------------------- /virtual_envs/deep3dface_env.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/virtual_envs/deep3dface_env.yaml -------------------------------------------------------------------------------- /virtual_envs/deepface_env.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/virtual_envs/deepface_env.yaml -------------------------------------------------------------------------------- /virtual_envs/mtcnn_env.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-gmpi/HEAD/virtual_envs/mtcnn_env.yaml --------------------------------------------------------------------------------