├── .gitignore ├── README.md ├── assets └── framework.png ├── configs └── vox1.yaml ├── data ├── README.md ├── fetch_data.sh ├── head_template.obj ├── landmark_embedding.npy ├── mean_deform.pt ├── transform.pkl └── u_full.pt ├── dataset ├── __init__.py ├── core.py └── video_data.py ├── decalib ├── __init__.py ├── deca.py ├── models │ ├── FLAME.py │ ├── decoders.py │ ├── encoders.py │ ├── frnet.py │ ├── lbs.py │ ├── resnet.py │ └── unet.py └── utils │ ├── config.py │ ├── lossfunc.py │ ├── rasterizer │ ├── INSTALL.md │ ├── __init__.py │ ├── setup.py │ ├── standard_rasterize_cuda.cpp │ └── standard_rasterize_cuda_kernel.cu │ ├── renderer.py │ ├── rotation_converter.py │ ├── tensor_cropper.py │ ├── trainer.py │ └── util.py ├── examples ├── 1.png ├── 2.png ├── exp.gif ├── jaw.gif ├── output.png ├── pose.gif └── shape.gif ├── inference.py ├── losses ├── __init__.py ├── perceptual.py ├── segmentation.py └── vggface.py ├── main_stage1.py ├── main_stage2.py ├── models ├── __init__.py ├── autoencoder.py ├── cvthead.py ├── discriminator.py ├── face_parsing.py ├── spiralnet.py ├── transformer.py ├── unet.py └── vgg2face.py ├── requirements.txt └── utils ├── checkpoint.py ├── common.py ├── logger.py ├── loss.py ├── meter.py ├── metrics.py ├── trainer.py └── visualize.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/README.md -------------------------------------------------------------------------------- /assets/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/assets/framework.png -------------------------------------------------------------------------------- /configs/vox1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/configs/vox1.yaml -------------------------------------------------------------------------------- /data/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/data/README.md -------------------------------------------------------------------------------- /data/fetch_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/data/fetch_data.sh -------------------------------------------------------------------------------- /data/head_template.obj: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/data/head_template.obj -------------------------------------------------------------------------------- /data/landmark_embedding.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/data/landmark_embedding.npy -------------------------------------------------------------------------------- /data/mean_deform.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/data/mean_deform.pt -------------------------------------------------------------------------------- /data/transform.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/data/transform.pkl -------------------------------------------------------------------------------- /data/u_full.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/data/u_full.pt -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- 1 | from dataset.core import worker_init_fn 2 | -------------------------------------------------------------------------------- /dataset/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/dataset/core.py -------------------------------------------------------------------------------- /dataset/video_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/dataset/video_data.py -------------------------------------------------------------------------------- /decalib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /decalib/deca.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/decalib/deca.py -------------------------------------------------------------------------------- /decalib/models/FLAME.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/decalib/models/FLAME.py -------------------------------------------------------------------------------- /decalib/models/decoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/decalib/models/decoders.py -------------------------------------------------------------------------------- /decalib/models/encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/decalib/models/encoders.py -------------------------------------------------------------------------------- /decalib/models/frnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/decalib/models/frnet.py -------------------------------------------------------------------------------- /decalib/models/lbs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/decalib/models/lbs.py -------------------------------------------------------------------------------- /decalib/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/decalib/models/resnet.py -------------------------------------------------------------------------------- /decalib/models/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/decalib/models/unet.py -------------------------------------------------------------------------------- /decalib/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/decalib/utils/config.py -------------------------------------------------------------------------------- /decalib/utils/lossfunc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/decalib/utils/lossfunc.py -------------------------------------------------------------------------------- /decalib/utils/rasterizer/INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/decalib/utils/rasterizer/INSTALL.md -------------------------------------------------------------------------------- /decalib/utils/rasterizer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /decalib/utils/rasterizer/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/decalib/utils/rasterizer/setup.py -------------------------------------------------------------------------------- /decalib/utils/rasterizer/standard_rasterize_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/decalib/utils/rasterizer/standard_rasterize_cuda.cpp -------------------------------------------------------------------------------- /decalib/utils/rasterizer/standard_rasterize_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/decalib/utils/rasterizer/standard_rasterize_cuda_kernel.cu -------------------------------------------------------------------------------- /decalib/utils/renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/decalib/utils/renderer.py -------------------------------------------------------------------------------- /decalib/utils/rotation_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/decalib/utils/rotation_converter.py -------------------------------------------------------------------------------- /decalib/utils/tensor_cropper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/decalib/utils/tensor_cropper.py -------------------------------------------------------------------------------- /decalib/utils/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/decalib/utils/trainer.py -------------------------------------------------------------------------------- /decalib/utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/decalib/utils/util.py -------------------------------------------------------------------------------- /examples/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/examples/1.png -------------------------------------------------------------------------------- /examples/2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/examples/2.png -------------------------------------------------------------------------------- /examples/exp.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/examples/exp.gif -------------------------------------------------------------------------------- /examples/jaw.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/examples/jaw.gif -------------------------------------------------------------------------------- /examples/output.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/examples/output.png -------------------------------------------------------------------------------- /examples/pose.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/examples/pose.gif -------------------------------------------------------------------------------- /examples/shape.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/examples/shape.gif -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/inference.py -------------------------------------------------------------------------------- /losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/losses/__init__.py -------------------------------------------------------------------------------- /losses/perceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/losses/perceptual.py -------------------------------------------------------------------------------- /losses/segmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/losses/segmentation.py -------------------------------------------------------------------------------- /losses/vggface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/losses/vggface.py -------------------------------------------------------------------------------- /main_stage1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/main_stage1.py -------------------------------------------------------------------------------- /main_stage2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/main_stage2.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/models/autoencoder.py -------------------------------------------------------------------------------- /models/cvthead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/models/cvthead.py -------------------------------------------------------------------------------- /models/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/models/discriminator.py -------------------------------------------------------------------------------- /models/face_parsing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/models/face_parsing.py -------------------------------------------------------------------------------- /models/spiralnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/models/spiralnet.py -------------------------------------------------------------------------------- /models/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/models/transformer.py -------------------------------------------------------------------------------- /models/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/models/unet.py -------------------------------------------------------------------------------- /models/vgg2face.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/models/vgg2face.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/requirements.txt -------------------------------------------------------------------------------- /utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/utils/checkpoint.py -------------------------------------------------------------------------------- /utils/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/utils/common.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/utils/loss.py -------------------------------------------------------------------------------- /utils/meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/utils/meter.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/utils/trainer.py -------------------------------------------------------------------------------- /utils/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HowieMa/CVTHead/HEAD/utils/visualize.py --------------------------------------------------------------------------------