├── .gitignore ├── EVALUATE.md ├── LICENSE ├── README.md ├── TRAINING.md ├── app.py ├── assets ├── animation.gif ├── screenshot.png ├── sith_demo.gif └── teaser.gif ├── checkpoints └── .gitkeep ├── data ├── body_models │ └── .gitkeep ├── examples │ ├── images │ │ ├── 000.png │ │ ├── 000_keypoints.json │ │ ├── 001.png │ │ ├── 001_keypoints.json │ │ ├── 002.png │ │ ├── 002_keypoints.json │ │ ├── 003.png │ │ ├── 003_keypoints.json │ │ ├── 004.png │ │ ├── 004_keypoints.json │ │ ├── 005.png │ │ ├── 005_keypoints.json │ │ ├── 006.png │ │ ├── 006_keypoints.json │ │ ├── 007.png │ │ ├── 007_keypoints.json │ │ ├── 008.png │ │ ├── 008_keypoints.json │ │ ├── 009.png │ │ ├── 009_keypoints.json │ │ ├── 010.png │ │ ├── 010_keypoints.json │ │ ├── 011.png │ │ ├── 011_keypoints.json │ │ ├── 012.png │ │ ├── 012_keypoints.json │ │ ├── 013.png │ │ ├── 013_keypoints.json │ │ ├── 014.png │ │ ├── 014_keypoints.json │ │ ├── 015.png │ │ └── 015_keypoints.json │ └── rgba │ │ ├── 000.png │ │ ├── 001.png │ │ ├── 002.png │ │ ├── 003.png │ │ ├── 004.png │ │ ├── 005.png │ │ ├── 006.png │ │ ├── 007.png │ │ ├── 008.png │ │ ├── 009.png │ │ ├── 010.png │ │ ├── 011.png │ │ ├── 012.png │ │ ├── 013.png │ │ ├── 014.png │ │ └── 015.png ├── gradio │ ├── pose_001.json │ ├── pose_001.png │ ├── pose_002.json │ ├── pose_002.png │ ├── pose_003.json │ ├── pose_003.png │ ├── pose_004.json │ ├── pose_004.png │ ├── pose_005.json │ ├── pose_005.png │ ├── pose_006.json │ ├── pose_006.png │ ├── pose_007.json │ ├── pose_007.png │ ├── pose_008.json │ ├── pose_008.png │ ├── pose_009.json │ ├── pose_009.png │ ├── pose_010.json │ ├── pose_010.png │ ├── pose_011.json │ ├── pose_011.png │ ├── pose_012.json │ ├── pose_012.png │ ├── pose_013.json │ ├── pose_013.png │ ├── pose_014.json │ ├── pose_014.png │ ├── pose_015.json │ ├── pose_015.png │ ├── pose_016.json │ └── pose_016.png ├── gradio_config.pkl ├── smplx_canonical.obj ├── smplx_openpose25.json ├── smplx_uv.obj └── smplx_watertight.pkl ├── demo ├── briarmbg.py ├── gradio_func.py └── load_obj.py ├── diffusion ├── README.md ├── config.yaml ├── lib │ ├── ccprojection.py │ ├── config.py │ ├── load_obj.py │ ├── pipeline.py │ ├── test_diffusion_dataset.py │ ├── train_diffusion_dataset.py │ └── utils.py └── train.py ├── fit.py ├── fitting ├── LICENSE ├── models │ ├── base_backbone.py │ ├── layer.py │ ├── smpler_x.py │ ├── smpler_x_model.py │ └── vit.py └── utils │ ├── LICENSE │ ├── conversion.py │ ├── kps.py │ └── transforms.py ├── hallucinate.py ├── recon ├── README.md ├── config.yaml ├── datasets │ ├── load_obj.py │ ├── test_dataset.py │ └── train_dataset.py ├── models │ ├── SMPL_query.py │ ├── evaluator.py │ ├── geo_model.py │ ├── networks │ │ ├── encoder.py │ │ ├── layers.py │ │ ├── mlps.py │ │ └── normal_predictor.py │ ├── nrm_trainer.py │ ├── ops │ │ └── mesh │ │ │ ├── __init__.py │ │ │ ├── area_weighted_distribution.py │ │ │ ├── barycentric_coordinates.py │ │ │ ├── closest_point.py │ │ │ ├── closest_tex.py │ │ │ ├── compute_sdf.py │ │ │ ├── load_obj.py │ │ │ ├── normalize.py │ │ │ ├── per_face_normals.py │ │ │ ├── per_vertex_normals.py │ │ │ ├── point_sample.py │ │ │ ├── random_face.py │ │ │ ├── sample_near_surface.py │ │ │ ├── sample_surface.py │ │ │ ├── sample_tex.py │ │ │ └── sample_uniform.py │ ├── positional_encoding.py │ ├── tex_model.py │ └── trainer.py ├── train_full.py ├── train_normal.py └── utils │ └── config.py ├── reconstruct.py ├── requirements.txt ├── run.sh └── tools ├── align_thuman.py ├── centralize_rgba.py ├── dataset_utils.py ├── download.sh ├── evaluate.py ├── evaluate_image.py └── load_smplx_json.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/.gitignore -------------------------------------------------------------------------------- /EVALUATE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/EVALUATE.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/README.md -------------------------------------------------------------------------------- /TRAINING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/TRAINING.md -------------------------------------------------------------------------------- /app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/app.py -------------------------------------------------------------------------------- /assets/animation.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/assets/animation.gif -------------------------------------------------------------------------------- /assets/screenshot.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/assets/screenshot.png -------------------------------------------------------------------------------- /assets/sith_demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/assets/sith_demo.gif -------------------------------------------------------------------------------- /assets/teaser.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/assets/teaser.gif -------------------------------------------------------------------------------- /checkpoints/.gitkeep: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /data/body_models/.gitkeep: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /data/examples/images/000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/000.png -------------------------------------------------------------------------------- /data/examples/images/000_keypoints.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/000_keypoints.json -------------------------------------------------------------------------------- /data/examples/images/001.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/001.png -------------------------------------------------------------------------------- /data/examples/images/001_keypoints.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/001_keypoints.json -------------------------------------------------------------------------------- /data/examples/images/002.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/002.png -------------------------------------------------------------------------------- /data/examples/images/002_keypoints.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/002_keypoints.json -------------------------------------------------------------------------------- /data/examples/images/003.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/003.png -------------------------------------------------------------------------------- /data/examples/images/003_keypoints.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/003_keypoints.json -------------------------------------------------------------------------------- /data/examples/images/004.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/004.png -------------------------------------------------------------------------------- /data/examples/images/004_keypoints.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/004_keypoints.json -------------------------------------------------------------------------------- /data/examples/images/005.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/005.png -------------------------------------------------------------------------------- /data/examples/images/005_keypoints.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/005_keypoints.json -------------------------------------------------------------------------------- /data/examples/images/006.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/006.png -------------------------------------------------------------------------------- /data/examples/images/006_keypoints.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/006_keypoints.json -------------------------------------------------------------------------------- /data/examples/images/007.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/007.png -------------------------------------------------------------------------------- /data/examples/images/007_keypoints.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/007_keypoints.json -------------------------------------------------------------------------------- /data/examples/images/008.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/008.png -------------------------------------------------------------------------------- /data/examples/images/008_keypoints.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/008_keypoints.json -------------------------------------------------------------------------------- /data/examples/images/009.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/009.png -------------------------------------------------------------------------------- /data/examples/images/009_keypoints.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/009_keypoints.json -------------------------------------------------------------------------------- /data/examples/images/010.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/010.png -------------------------------------------------------------------------------- /data/examples/images/010_keypoints.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/010_keypoints.json -------------------------------------------------------------------------------- /data/examples/images/011.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/011.png -------------------------------------------------------------------------------- /data/examples/images/011_keypoints.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/011_keypoints.json -------------------------------------------------------------------------------- /data/examples/images/012.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/012.png -------------------------------------------------------------------------------- /data/examples/images/012_keypoints.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/012_keypoints.json -------------------------------------------------------------------------------- /data/examples/images/013.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/013.png -------------------------------------------------------------------------------- /data/examples/images/013_keypoints.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/013_keypoints.json -------------------------------------------------------------------------------- /data/examples/images/014.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/014.png -------------------------------------------------------------------------------- /data/examples/images/014_keypoints.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/014_keypoints.json -------------------------------------------------------------------------------- /data/examples/images/015.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/015.png -------------------------------------------------------------------------------- /data/examples/images/015_keypoints.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/images/015_keypoints.json -------------------------------------------------------------------------------- /data/examples/rgba/000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/rgba/000.png -------------------------------------------------------------------------------- /data/examples/rgba/001.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/rgba/001.png -------------------------------------------------------------------------------- /data/examples/rgba/002.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/rgba/002.png -------------------------------------------------------------------------------- /data/examples/rgba/003.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/rgba/003.png -------------------------------------------------------------------------------- /data/examples/rgba/004.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/rgba/004.png -------------------------------------------------------------------------------- /data/examples/rgba/005.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/rgba/005.png -------------------------------------------------------------------------------- /data/examples/rgba/006.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/rgba/006.png -------------------------------------------------------------------------------- /data/examples/rgba/007.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/rgba/007.png -------------------------------------------------------------------------------- /data/examples/rgba/008.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/rgba/008.png -------------------------------------------------------------------------------- /data/examples/rgba/009.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/rgba/009.png -------------------------------------------------------------------------------- /data/examples/rgba/010.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/rgba/010.png -------------------------------------------------------------------------------- /data/examples/rgba/011.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/rgba/011.png -------------------------------------------------------------------------------- /data/examples/rgba/012.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/rgba/012.png -------------------------------------------------------------------------------- /data/examples/rgba/013.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/rgba/013.png -------------------------------------------------------------------------------- /data/examples/rgba/014.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/rgba/014.png -------------------------------------------------------------------------------- /data/examples/rgba/015.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/examples/rgba/015.png -------------------------------------------------------------------------------- /data/gradio/pose_001.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_001.json -------------------------------------------------------------------------------- /data/gradio/pose_001.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_001.png -------------------------------------------------------------------------------- /data/gradio/pose_002.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_002.json -------------------------------------------------------------------------------- /data/gradio/pose_002.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_002.png -------------------------------------------------------------------------------- /data/gradio/pose_003.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_003.json -------------------------------------------------------------------------------- /data/gradio/pose_003.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_003.png -------------------------------------------------------------------------------- /data/gradio/pose_004.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_004.json -------------------------------------------------------------------------------- /data/gradio/pose_004.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_004.png -------------------------------------------------------------------------------- /data/gradio/pose_005.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_005.json -------------------------------------------------------------------------------- /data/gradio/pose_005.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_005.png -------------------------------------------------------------------------------- /data/gradio/pose_006.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_006.json -------------------------------------------------------------------------------- /data/gradio/pose_006.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_006.png -------------------------------------------------------------------------------- /data/gradio/pose_007.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_007.json -------------------------------------------------------------------------------- /data/gradio/pose_007.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_007.png -------------------------------------------------------------------------------- /data/gradio/pose_008.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_008.json -------------------------------------------------------------------------------- /data/gradio/pose_008.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_008.png -------------------------------------------------------------------------------- /data/gradio/pose_009.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_009.json -------------------------------------------------------------------------------- /data/gradio/pose_009.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_009.png -------------------------------------------------------------------------------- /data/gradio/pose_010.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_010.json -------------------------------------------------------------------------------- /data/gradio/pose_010.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_010.png -------------------------------------------------------------------------------- /data/gradio/pose_011.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_011.json -------------------------------------------------------------------------------- /data/gradio/pose_011.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_011.png -------------------------------------------------------------------------------- /data/gradio/pose_012.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_012.json -------------------------------------------------------------------------------- /data/gradio/pose_012.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_012.png -------------------------------------------------------------------------------- /data/gradio/pose_013.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_013.json -------------------------------------------------------------------------------- /data/gradio/pose_013.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_013.png -------------------------------------------------------------------------------- /data/gradio/pose_014.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_014.json -------------------------------------------------------------------------------- /data/gradio/pose_014.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_014.png -------------------------------------------------------------------------------- /data/gradio/pose_015.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_015.json -------------------------------------------------------------------------------- /data/gradio/pose_015.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_015.png -------------------------------------------------------------------------------- /data/gradio/pose_016.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_016.json -------------------------------------------------------------------------------- /data/gradio/pose_016.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio/pose_016.png -------------------------------------------------------------------------------- /data/gradio_config.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/gradio_config.pkl -------------------------------------------------------------------------------- /data/smplx_canonical.obj: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/smplx_canonical.obj -------------------------------------------------------------------------------- /data/smplx_openpose25.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/smplx_openpose25.json -------------------------------------------------------------------------------- /data/smplx_uv.obj: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/smplx_uv.obj -------------------------------------------------------------------------------- /data/smplx_watertight.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/data/smplx_watertight.pkl -------------------------------------------------------------------------------- /demo/briarmbg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/demo/briarmbg.py -------------------------------------------------------------------------------- /demo/gradio_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/demo/gradio_func.py -------------------------------------------------------------------------------- /demo/load_obj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/demo/load_obj.py -------------------------------------------------------------------------------- /diffusion/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/diffusion/README.md -------------------------------------------------------------------------------- /diffusion/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/diffusion/config.yaml -------------------------------------------------------------------------------- /diffusion/lib/ccprojection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/diffusion/lib/ccprojection.py -------------------------------------------------------------------------------- /diffusion/lib/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/diffusion/lib/config.py -------------------------------------------------------------------------------- /diffusion/lib/load_obj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/diffusion/lib/load_obj.py -------------------------------------------------------------------------------- /diffusion/lib/pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/diffusion/lib/pipeline.py -------------------------------------------------------------------------------- /diffusion/lib/test_diffusion_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/diffusion/lib/test_diffusion_dataset.py -------------------------------------------------------------------------------- /diffusion/lib/train_diffusion_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/diffusion/lib/train_diffusion_dataset.py -------------------------------------------------------------------------------- /diffusion/lib/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/diffusion/lib/utils.py -------------------------------------------------------------------------------- /diffusion/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/diffusion/train.py -------------------------------------------------------------------------------- /fit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/fit.py -------------------------------------------------------------------------------- /fitting/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/fitting/LICENSE -------------------------------------------------------------------------------- /fitting/models/base_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/fitting/models/base_backbone.py -------------------------------------------------------------------------------- /fitting/models/layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/fitting/models/layer.py -------------------------------------------------------------------------------- /fitting/models/smpler_x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/fitting/models/smpler_x.py -------------------------------------------------------------------------------- /fitting/models/smpler_x_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/fitting/models/smpler_x_model.py -------------------------------------------------------------------------------- /fitting/models/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/fitting/models/vit.py -------------------------------------------------------------------------------- /fitting/utils/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/fitting/utils/LICENSE -------------------------------------------------------------------------------- /fitting/utils/conversion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/fitting/utils/conversion.py -------------------------------------------------------------------------------- /fitting/utils/kps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/fitting/utils/kps.py -------------------------------------------------------------------------------- /fitting/utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/fitting/utils/transforms.py -------------------------------------------------------------------------------- /hallucinate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/hallucinate.py -------------------------------------------------------------------------------- /recon/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/README.md -------------------------------------------------------------------------------- /recon/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/config.yaml -------------------------------------------------------------------------------- /recon/datasets/load_obj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/datasets/load_obj.py -------------------------------------------------------------------------------- /recon/datasets/test_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/datasets/test_dataset.py -------------------------------------------------------------------------------- /recon/datasets/train_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/datasets/train_dataset.py -------------------------------------------------------------------------------- /recon/models/SMPL_query.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/models/SMPL_query.py -------------------------------------------------------------------------------- /recon/models/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/models/evaluator.py -------------------------------------------------------------------------------- /recon/models/geo_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/models/geo_model.py -------------------------------------------------------------------------------- /recon/models/networks/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/models/networks/encoder.py -------------------------------------------------------------------------------- /recon/models/networks/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/models/networks/layers.py -------------------------------------------------------------------------------- /recon/models/networks/mlps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/models/networks/mlps.py -------------------------------------------------------------------------------- /recon/models/networks/normal_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/models/networks/normal_predictor.py -------------------------------------------------------------------------------- /recon/models/nrm_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/models/nrm_trainer.py -------------------------------------------------------------------------------- /recon/models/ops/mesh/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/models/ops/mesh/__init__.py -------------------------------------------------------------------------------- /recon/models/ops/mesh/area_weighted_distribution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/models/ops/mesh/area_weighted_distribution.py -------------------------------------------------------------------------------- /recon/models/ops/mesh/barycentric_coordinates.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/models/ops/mesh/barycentric_coordinates.py -------------------------------------------------------------------------------- /recon/models/ops/mesh/closest_point.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/models/ops/mesh/closest_point.py -------------------------------------------------------------------------------- /recon/models/ops/mesh/closest_tex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/models/ops/mesh/closest_tex.py -------------------------------------------------------------------------------- /recon/models/ops/mesh/compute_sdf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/models/ops/mesh/compute_sdf.py -------------------------------------------------------------------------------- /recon/models/ops/mesh/load_obj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/models/ops/mesh/load_obj.py -------------------------------------------------------------------------------- /recon/models/ops/mesh/normalize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/models/ops/mesh/normalize.py -------------------------------------------------------------------------------- /recon/models/ops/mesh/per_face_normals.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/models/ops/mesh/per_face_normals.py -------------------------------------------------------------------------------- /recon/models/ops/mesh/per_vertex_normals.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/models/ops/mesh/per_vertex_normals.py -------------------------------------------------------------------------------- /recon/models/ops/mesh/point_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/models/ops/mesh/point_sample.py -------------------------------------------------------------------------------- /recon/models/ops/mesh/random_face.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/models/ops/mesh/random_face.py -------------------------------------------------------------------------------- /recon/models/ops/mesh/sample_near_surface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/models/ops/mesh/sample_near_surface.py -------------------------------------------------------------------------------- /recon/models/ops/mesh/sample_surface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/models/ops/mesh/sample_surface.py -------------------------------------------------------------------------------- /recon/models/ops/mesh/sample_tex.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/models/ops/mesh/sample_tex.py -------------------------------------------------------------------------------- /recon/models/ops/mesh/sample_uniform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/models/ops/mesh/sample_uniform.py -------------------------------------------------------------------------------- /recon/models/positional_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/models/positional_encoding.py -------------------------------------------------------------------------------- /recon/models/tex_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/models/tex_model.py -------------------------------------------------------------------------------- /recon/models/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/models/trainer.py -------------------------------------------------------------------------------- /recon/train_full.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/train_full.py -------------------------------------------------------------------------------- /recon/train_normal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/train_normal.py -------------------------------------------------------------------------------- /recon/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/recon/utils/config.py -------------------------------------------------------------------------------- /reconstruct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/reconstruct.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/run.sh -------------------------------------------------------------------------------- /tools/align_thuman.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/tools/align_thuman.py -------------------------------------------------------------------------------- /tools/centralize_rgba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/tools/centralize_rgba.py -------------------------------------------------------------------------------- /tools/dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/tools/dataset_utils.py -------------------------------------------------------------------------------- /tools/download.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/tools/download.sh -------------------------------------------------------------------------------- /tools/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/tools/evaluate.py -------------------------------------------------------------------------------- /tools/evaluate_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/tools/evaluate_image.py -------------------------------------------------------------------------------- /tools/load_smplx_json.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SiTH-Diffusion/SiTH/HEAD/tools/load_smplx_json.py --------------------------------------------------------------------------------