├── README.md ├── assets ├── pipeline-2.png └── test ├── code ├── cam_utils.py ├── dataset.py ├── networks │ ├── __init__.py │ ├── encoder3d.py │ └── headnerf.py ├── pretrained_models │ └── eg3d │ │ └── test ├── run_recon_video_3dmm.py ├── run_recon_video_audio.py ├── run_recon_video_rgb.py ├── test ├── train_3dmm.py ├── train_audio.py ├── train_rgb.py ├── trainer_3dmm.py ├── trainer_audio.py └── trainer_rgb.py └── eg3d-pose-detection ├── 3dface2idr.py ├── batch_mtcnn.py ├── camera2label.py ├── crop_images.py ├── data ├── __init__.py ├── base_dataset.py ├── flist_dataset.py ├── image_folder.py └── template_dataset.py ├── models ├── __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_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 │ ├── requirement.txt │ ├── run.sh │ ├── torch2onnx.py │ ├── train.py │ └── utils │ │ ├── __init__.py │ │ ├── plot.py │ │ ├── utils_callbacks.py │ │ ├── utils_config.py │ │ ├── utils_distributed_sampler.py │ │ └── utils_logging.py ├── base_model.py ├── bfm.py ├── facerecon_model.py ├── losses.py ├── networks.py └── template_model.py ├── options ├── __init__.py ├── base_options.py ├── test_options.py └── train_options.py ├── process_test_video.py ├── smooth.py ├── test └── test.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/README.md -------------------------------------------------------------------------------- /assets/pipeline-2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/assets/pipeline-2.png -------------------------------------------------------------------------------- /assets/test: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /code/cam_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/code/cam_utils.py -------------------------------------------------------------------------------- /code/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/code/dataset.py -------------------------------------------------------------------------------- /code/networks/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /code/networks/encoder3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/code/networks/encoder3d.py -------------------------------------------------------------------------------- /code/networks/headnerf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/code/networks/headnerf.py -------------------------------------------------------------------------------- /code/pretrained_models/eg3d/test: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /code/run_recon_video_3dmm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/code/run_recon_video_3dmm.py -------------------------------------------------------------------------------- /code/run_recon_video_audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/code/run_recon_video_audio.py -------------------------------------------------------------------------------- /code/run_recon_video_rgb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/code/run_recon_video_rgb.py -------------------------------------------------------------------------------- /code/test: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /code/train_3dmm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/code/train_3dmm.py -------------------------------------------------------------------------------- /code/train_audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/code/train_audio.py -------------------------------------------------------------------------------- /code/train_rgb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/code/train_rgb.py -------------------------------------------------------------------------------- /code/trainer_3dmm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/code/trainer_3dmm.py -------------------------------------------------------------------------------- /code/trainer_audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/code/trainer_audio.py -------------------------------------------------------------------------------- /code/trainer_rgb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/code/trainer_rgb.py -------------------------------------------------------------------------------- /eg3d-pose-detection/3dface2idr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/3dface2idr.py -------------------------------------------------------------------------------- /eg3d-pose-detection/batch_mtcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/batch_mtcnn.py -------------------------------------------------------------------------------- /eg3d-pose-detection/camera2label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/camera2label.py -------------------------------------------------------------------------------- /eg3d-pose-detection/crop_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/crop_images.py -------------------------------------------------------------------------------- /eg3d-pose-detection/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/data/__init__.py -------------------------------------------------------------------------------- /eg3d-pose-detection/data/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/data/base_dataset.py -------------------------------------------------------------------------------- /eg3d-pose-detection/data/flist_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/data/flist_dataset.py -------------------------------------------------------------------------------- /eg3d-pose-detection/data/image_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/data/image_folder.py -------------------------------------------------------------------------------- /eg3d-pose-detection/data/template_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/data/template_dataset.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/__init__.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/README.md -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/backbones/__init__.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/backbones/iresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/backbones/iresnet.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/backbones/iresnet2060.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/backbones/iresnet2060.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/backbones/mobilefacenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/backbones/mobilefacenet.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/backbones/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/backbones/vit.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/3millions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/3millions.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/base.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/glint360k_mbf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/glint360k_mbf.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/glint360k_r100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/glint360k_r100.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/glint360k_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/glint360k_r50.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/ms1mv2_mbf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/ms1mv2_mbf.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/ms1mv2_r100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/ms1mv2_r100.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/ms1mv2_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/ms1mv2_r50.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/ms1mv3_mbf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/ms1mv3_mbf.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/ms1mv3_r100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/ms1mv3_r100.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/ms1mv3_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/ms1mv3_r50.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf12m_conflict_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf12m_conflict_r50.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf12m_conflict_r50_pfc03_filter04.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf12m_conflict_r50_pfc03_filter04.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf12m_flip_pfc01_filter04_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf12m_flip_pfc01_filter04_r50.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf12m_flip_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf12m_flip_r50.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf12m_mbf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf12m_mbf.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf12m_pfc02_r100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf12m_pfc02_r100.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf12m_r100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf12m_r100.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf12m_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf12m_r50.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc0008_32gpu_r100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc0008_32gpu_r100.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc02_16gpus_mbf_bs8k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc02_16gpus_mbf_bs8k.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc02_16gpus_r100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc02_16gpus_r100.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc02_16gpus_r50_bs8k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc02_16gpus_r50_bs8k.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc02_32gpus_r50_bs4k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc02_32gpus_r50_bs4k.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc02_8gpus_r50_bs4k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc02_8gpus_r50_bs4k.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc02_r100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc02_r100.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc02_r100_16gpus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc02_r100_16gpus.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc02_r100_32gpus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc02_r100_32gpus.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc03_32gpu_r100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc03_32gpu_r100.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc03_32gpu_r18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc03_32gpu_r18.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc03_32gpu_r200.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc03_32gpu_r200.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc03_32gpu_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc03_32gpu_r50.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc03_40epoch_64gpu_vit_b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc03_40epoch_64gpu_vit_b.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc03_40epoch_64gpu_vit_l.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc03_40epoch_64gpu_vit_l.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc03_40epoch_64gpu_vit_s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc03_40epoch_64gpu_vit_s.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc03_40epoch_64gpu_vit_t.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc03_40epoch_64gpu_vit_t.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc03_40epoch_8gpu_vit_t.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf42m_pfc03_40epoch_8gpu_vit_t.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf4m_mbf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf4m_mbf.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf4m_r100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf4m_r100.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/configs/wf4m_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/configs/wf4m_r50.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/dataset.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/dist.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/dist.sh -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/docs/eval.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/docs/eval.md -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/docs/install.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/docs/install.md -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/docs/install_dali.md: -------------------------------------------------------------------------------- 1 | TODO 2 | -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/docs/modelzoo.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/docs/prepare_webface42m.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/docs/prepare_webface42m.md -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/docs/speed_benchmark.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/docs/speed_benchmark.md -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/eval/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/eval/verification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/eval/verification.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/eval_ijbc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/eval_ijbc.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/flops.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/inference.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/losses.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/lr_scheduler.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/onnx_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/onnx_helper.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/onnx_ijbc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/onnx_ijbc.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/partial_fc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/partial_fc.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/requirement.txt: -------------------------------------------------------------------------------- 1 | tensorboard 2 | easydict 3 | mxnet 4 | onnx 5 | sklearn 6 | -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/run.sh -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/torch2onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/torch2onnx.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/train.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/utils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/utils/plot.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/utils/utils_callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/utils/utils_callbacks.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/utils/utils_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/utils/utils_config.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/utils/utils_distributed_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/utils/utils_distributed_sampler.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/arcface_torch/utils/utils_logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/arcface_torch/utils/utils_logging.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/base_model.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/bfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/bfm.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/facerecon_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/facerecon_model.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/losses.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/networks.py -------------------------------------------------------------------------------- /eg3d-pose-detection/models/template_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/models/template_model.py -------------------------------------------------------------------------------- /eg3d-pose-detection/options/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/options/__init__.py -------------------------------------------------------------------------------- /eg3d-pose-detection/options/base_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/options/base_options.py -------------------------------------------------------------------------------- /eg3d-pose-detection/options/test_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/options/test_options.py -------------------------------------------------------------------------------- /eg3d-pose-detection/options/train_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/options/train_options.py -------------------------------------------------------------------------------- /eg3d-pose-detection/process_test_video.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/process_test_video.py -------------------------------------------------------------------------------- /eg3d-pose-detection/smooth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/smooth.py -------------------------------------------------------------------------------- /eg3d-pose-detection/test: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /eg3d-pose-detection/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bbaaii/HFA-GP/HEAD/eg3d-pose-detection/test.py --------------------------------------------------------------------------------