├── .gitignore ├── LICENSE ├── README.md ├── assets ├── framework.png └── teaser.png ├── loss ├── __init__.py ├── gan_loss.py └── losses.py ├── model ├── __init__.py ├── binary_extractor.py ├── d3dfr │ ├── BFM │ │ └── std_exp.txt │ ├── __init__.py │ ├── arcface_torch │ │ ├── README.md │ │ ├── backbones │ │ │ ├── __init__.py │ │ │ ├── iresnet.py │ │ │ ├── iresnet2060.py │ │ │ ├── mobilefacenet.py │ │ │ └── vit.py │ │ ├── configs │ │ │ ├── 3millions.py │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── glint360k_mbf.py │ │ │ ├── glint360k_r100.py │ │ │ ├── glint360k_r50.py │ │ │ ├── ms1mv2_mbf.py │ │ │ ├── ms1mv2_r100.py │ │ │ ├── ms1mv2_r50.py │ │ │ ├── ms1mv3_mbf.py │ │ │ ├── ms1mv3_r100.py │ │ │ ├── ms1mv3_r50.py │ │ │ ├── wf12m_conflict_r50.py │ │ │ ├── wf12m_conflict_r50_pfc03_filter04.py │ │ │ ├── wf12m_flip_pfc01_filter04_r50.py │ │ │ ├── wf12m_flip_r50.py │ │ │ ├── wf12m_mbf.py │ │ │ ├── wf12m_pfc02_r100.py │ │ │ ├── wf12m_r100.py │ │ │ ├── wf12m_r50.py │ │ │ ├── wf42m_pfc0008_32gpu_r100.py │ │ │ ├── wf42m_pfc02_16gpus_mbf_bs8k.py │ │ │ ├── wf42m_pfc02_16gpus_r100.py │ │ │ ├── wf42m_pfc02_16gpus_r50_bs8k.py │ │ │ ├── wf42m_pfc02_32gpus_r50_bs4k.py │ │ │ ├── wf42m_pfc02_8gpus_r50_bs4k.py │ │ │ ├── wf42m_pfc02_r100.py │ │ │ ├── wf42m_pfc02_r100_16gpus.py │ │ │ ├── wf42m_pfc02_r100_32gpus.py │ │ │ ├── wf42m_pfc03_32gpu_r100.py │ │ │ ├── wf42m_pfc03_32gpu_r18.py │ │ │ ├── wf42m_pfc03_32gpu_r200.py │ │ │ ├── wf42m_pfc03_32gpu_r50.py │ │ │ ├── wf42m_pfc03_40epoch_64gpu_vit_b.py │ │ │ ├── wf42m_pfc03_40epoch_64gpu_vit_l.py │ │ │ ├── wf42m_pfc03_40epoch_64gpu_vit_s.py │ │ │ ├── wf42m_pfc03_40epoch_64gpu_vit_t.py │ │ │ ├── wf42m_pfc03_40epoch_8gpu_vit_b.py │ │ │ ├── wf42m_pfc03_40epoch_8gpu_vit_t.py │ │ │ ├── wf4m_mbf.py │ │ │ ├── wf4m_r100.py │ │ │ └── wf4m_r50.py │ │ ├── dataset.py │ │ ├── dist.sh │ │ ├── docs │ │ │ ├── eval.md │ │ │ ├── install.md │ │ │ ├── install_dali.md │ │ │ ├── modelzoo.md │ │ │ ├── prepare_webface42m.md │ │ │ └── speed_benchmark.md │ │ ├── eval │ │ │ ├── __init__.py │ │ │ └── verification.py │ │ ├── eval_ijbc.py │ │ ├── flops.py │ │ ├── inference.py │ │ ├── losses.py │ │ ├── lr_scheduler.py │ │ ├── onnx_helper.py │ │ ├── onnx_ijbc.py │ │ ├── partial_fc.py │ │ ├── partial_fc_v2.py │ │ ├── requirement.txt │ │ ├── run.sh │ │ ├── torch2onnx.py │ │ ├── train.py │ │ ├── train_v2.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── plot.py │ │ │ ├── utils_callbacks.py │ │ │ ├── utils_config.py │ │ │ ├── utils_distributed_sampler.py │ │ │ └── utils_logging.py │ ├── base_model.py │ ├── bfm.py │ ├── face_reconstruct_model.py │ ├── networks.py │ ├── renderer.py │ └── utils_3dmm │ │ ├── __init__.py │ │ ├── html.py │ │ ├── load_mats.py │ │ ├── preprocess.py │ │ ├── util.py │ │ └── visualizer.py ├── mlp.py └── style │ ├── non_leaking.py │ ├── op │ ├── __init__.py │ ├── conv2d_gradfix.py │ ├── fused_act.py │ ├── fused_bias_act.cpp │ ├── fused_bias_act_kernel.cu │ ├── upfirdn2d.cpp │ ├── upfirdn2d.py │ └── upfirdn2d_kernel.cu │ └── style_adaptive_mask.py ├── option.py ├── requirements.txt ├── test.py ├── train.py ├── train.sh ├── utils ├── __init__.py ├── binary_converter.py ├── dataset.py ├── distributed.py ├── logger.py ├── operate.py ├── util.py └── warmup_scheduler.py └── weights └── README.md /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/README.md -------------------------------------------------------------------------------- /assets/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/assets/framework.png -------------------------------------------------------------------------------- /assets/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/assets/teaser.png -------------------------------------------------------------------------------- /loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/loss/__init__.py -------------------------------------------------------------------------------- /loss/gan_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/loss/gan_loss.py -------------------------------------------------------------------------------- /loss/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/loss/losses.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/binary_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/binary_extractor.py -------------------------------------------------------------------------------- /model/d3dfr/BFM/std_exp.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/BFM/std_exp.txt -------------------------------------------------------------------------------- /model/d3dfr/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/__init__.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/README.md -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/backbones/__init__.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/backbones/iresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/backbones/iresnet.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/backbones/iresnet2060.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/backbones/iresnet2060.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/backbones/mobilefacenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/backbones/mobilefacenet.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/backbones/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/backbones/vit.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/3millions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/3millions.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/base.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/glint360k_mbf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/glint360k_mbf.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/glint360k_r100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/glint360k_r100.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/glint360k_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/glint360k_r50.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/ms1mv2_mbf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/ms1mv2_mbf.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/ms1mv2_r100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/ms1mv2_r100.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/ms1mv2_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/ms1mv2_r50.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/ms1mv3_mbf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/ms1mv3_mbf.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/ms1mv3_r100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/ms1mv3_r100.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/ms1mv3_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/ms1mv3_r50.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf12m_conflict_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf12m_conflict_r50.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf12m_conflict_r50_pfc03_filter04.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf12m_conflict_r50_pfc03_filter04.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf12m_flip_pfc01_filter04_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf12m_flip_pfc01_filter04_r50.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf12m_flip_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf12m_flip_r50.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf12m_mbf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf12m_mbf.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf12m_pfc02_r100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf12m_pfc02_r100.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf12m_r100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf12m_r100.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf12m_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf12m_r50.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf42m_pfc0008_32gpu_r100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf42m_pfc0008_32gpu_r100.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf42m_pfc02_16gpus_mbf_bs8k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf42m_pfc02_16gpus_mbf_bs8k.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf42m_pfc02_16gpus_r100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf42m_pfc02_16gpus_r100.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf42m_pfc02_16gpus_r50_bs8k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf42m_pfc02_16gpus_r50_bs8k.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf42m_pfc02_32gpus_r50_bs4k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf42m_pfc02_32gpus_r50_bs4k.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf42m_pfc02_8gpus_r50_bs4k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf42m_pfc02_8gpus_r50_bs4k.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf42m_pfc02_r100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf42m_pfc02_r100.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf42m_pfc02_r100_16gpus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf42m_pfc02_r100_16gpus.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf42m_pfc02_r100_32gpus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf42m_pfc02_r100_32gpus.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf42m_pfc03_32gpu_r100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf42m_pfc03_32gpu_r100.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf42m_pfc03_32gpu_r18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf42m_pfc03_32gpu_r18.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf42m_pfc03_32gpu_r200.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf42m_pfc03_32gpu_r200.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf42m_pfc03_32gpu_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf42m_pfc03_32gpu_r50.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf42m_pfc03_40epoch_64gpu_vit_b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf42m_pfc03_40epoch_64gpu_vit_b.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf42m_pfc03_40epoch_64gpu_vit_l.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf42m_pfc03_40epoch_64gpu_vit_l.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf42m_pfc03_40epoch_64gpu_vit_s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf42m_pfc03_40epoch_64gpu_vit_s.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf42m_pfc03_40epoch_64gpu_vit_t.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf42m_pfc03_40epoch_64gpu_vit_t.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf42m_pfc03_40epoch_8gpu_vit_b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf42m_pfc03_40epoch_8gpu_vit_b.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf42m_pfc03_40epoch_8gpu_vit_t.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf42m_pfc03_40epoch_8gpu_vit_t.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf4m_mbf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf4m_mbf.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf4m_r100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf4m_r100.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/configs/wf4m_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/configs/wf4m_r50.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/dataset.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/dist.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/dist.sh -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/docs/eval.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/docs/eval.md -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/docs/install.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/docs/install.md -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/docs/install_dali.md: -------------------------------------------------------------------------------- 1 | TODO 2 | -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/docs/modelzoo.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/docs/prepare_webface42m.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/docs/prepare_webface42m.md -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/docs/speed_benchmark.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/docs/speed_benchmark.md -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/eval/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/eval/verification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/eval/verification.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/eval_ijbc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/eval_ijbc.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/flops.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/inference.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/losses.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/lr_scheduler.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/onnx_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/onnx_helper.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/onnx_ijbc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/onnx_ijbc.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/partial_fc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/partial_fc.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/partial_fc_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/partial_fc_v2.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/requirement.txt: -------------------------------------------------------------------------------- 1 | tensorboard 2 | easydict 3 | mxnet 4 | onnx 5 | sklearn 6 | -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/run.sh -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/torch2onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/torch2onnx.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/train.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/train_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/train_v2.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/utils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/utils/plot.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/utils/utils_callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/utils/utils_callbacks.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/utils/utils_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/utils/utils_config.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/utils/utils_distributed_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/utils/utils_distributed_sampler.py -------------------------------------------------------------------------------- /model/d3dfr/arcface_torch/utils/utils_logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/arcface_torch/utils/utils_logging.py -------------------------------------------------------------------------------- /model/d3dfr/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/base_model.py -------------------------------------------------------------------------------- /model/d3dfr/bfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/bfm.py -------------------------------------------------------------------------------- /model/d3dfr/face_reconstruct_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/face_reconstruct_model.py -------------------------------------------------------------------------------- /model/d3dfr/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/networks.py -------------------------------------------------------------------------------- /model/d3dfr/renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/renderer.py -------------------------------------------------------------------------------- /model/d3dfr/utils_3dmm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/utils_3dmm/__init__.py -------------------------------------------------------------------------------- /model/d3dfr/utils_3dmm/html.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/utils_3dmm/html.py -------------------------------------------------------------------------------- /model/d3dfr/utils_3dmm/load_mats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/utils_3dmm/load_mats.py -------------------------------------------------------------------------------- /model/d3dfr/utils_3dmm/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/utils_3dmm/preprocess.py -------------------------------------------------------------------------------- /model/d3dfr/utils_3dmm/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/utils_3dmm/util.py -------------------------------------------------------------------------------- /model/d3dfr/utils_3dmm/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/d3dfr/utils_3dmm/visualizer.py -------------------------------------------------------------------------------- /model/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/mlp.py -------------------------------------------------------------------------------- /model/style/non_leaking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/style/non_leaking.py -------------------------------------------------------------------------------- /model/style/op/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/style/op/__init__.py -------------------------------------------------------------------------------- /model/style/op/conv2d_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/style/op/conv2d_gradfix.py -------------------------------------------------------------------------------- /model/style/op/fused_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/style/op/fused_act.py -------------------------------------------------------------------------------- /model/style/op/fused_bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/style/op/fused_bias_act.cpp -------------------------------------------------------------------------------- /model/style/op/fused_bias_act_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/style/op/fused_bias_act_kernel.cu -------------------------------------------------------------------------------- /model/style/op/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/style/op/upfirdn2d.cpp -------------------------------------------------------------------------------- /model/style/op/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/style/op/upfirdn2d.py -------------------------------------------------------------------------------- /model/style/op/upfirdn2d_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/style/op/upfirdn2d_kernel.cu -------------------------------------------------------------------------------- /model/style/style_adaptive_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/model/style/style_adaptive_mask.py -------------------------------------------------------------------------------- /option.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/option.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/requirements.txt -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/train.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/train.sh -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/binary_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/utils/binary_converter.py -------------------------------------------------------------------------------- /utils/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/utils/dataset.py -------------------------------------------------------------------------------- /utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/utils/distributed.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/operate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/utils/operate.py -------------------------------------------------------------------------------- /utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/utils/util.py -------------------------------------------------------------------------------- /utils/warmup_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Harxis/G2Face/HEAD/utils/warmup_scheduler.py -------------------------------------------------------------------------------- /weights/README.md: -------------------------------------------------------------------------------- 1 | path to pretrained weights --------------------------------------------------------------------------------