├── .gitmodules ├── LICENSE ├── README.md ├── configs ├── config.py ├── constants.py └── yamls │ ├── demo.yaml │ ├── model_base.yaml │ ├── stage1.yaml │ ├── stage2.yaml │ └── stage2_b.yaml ├── demo.py ├── docs ├── API.md ├── DATASET.md ├── DOCKER.md └── INSTALL.md ├── fetch_demo_data.sh ├── lib ├── core │ ├── loss.py │ └── trainer.py ├── data │ ├── __init__.py │ ├── _dataset.py │ ├── dataloader.py │ ├── datasets │ │ ├── __init__.py │ │ ├── amass.py │ │ ├── bedlam.py │ │ ├── dataset2d.py │ │ ├── dataset3d.py │ │ ├── dataset_custom.py │ │ ├── dataset_eval.py │ │ ├── mixed_dataset.py │ │ └── videos.py │ └── utils │ │ ├── augmentor.py │ │ └── normalizer.py ├── data_utils │ ├── amass_utils.py │ ├── emdb_eval_utils.py │ ├── rich_eval_utils.py │ ├── threedpw_eval_utils.py │ └── threedpw_train_utils.py ├── eval │ ├── eval_utils.py │ ├── evaluate_3dpw.py │ ├── evaluate_emdb.py │ └── evaluate_rich.py ├── models │ ├── __init__.py │ ├── layers │ │ ├── __init__.py │ │ ├── modules.py │ │ └── utils.py │ ├── preproc │ │ ├── backbone │ │ │ ├── hmr2.py │ │ │ ├── pose_transformer.py │ │ │ ├── smpl_head.py │ │ │ ├── t_cond_mlp.py │ │ │ ├── utils.py │ │ │ └── vit.py │ │ ├── detector.py │ │ ├── extractor.py │ │ └── slam.py │ ├── smpl.py │ ├── smplify │ │ ├── __init__.py │ │ ├── losses.py │ │ └── smplify.py │ └── wham.py ├── utils │ ├── data_utils.py │ ├── imutils.py │ ├── kp_utils.py │ ├── transforms.py │ └── utils.py └── vis │ ├── renderer.py │ ├── run_vis.py │ └── tools.py ├── requirements.txt ├── train.py └── wham_api.py /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/.gitmodules -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/README.md -------------------------------------------------------------------------------- /configs/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/configs/config.py -------------------------------------------------------------------------------- /configs/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/configs/constants.py -------------------------------------------------------------------------------- /configs/yamls/demo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/configs/yamls/demo.yaml -------------------------------------------------------------------------------- /configs/yamls/model_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/configs/yamls/model_base.yaml -------------------------------------------------------------------------------- /configs/yamls/stage1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/configs/yamls/stage1.yaml -------------------------------------------------------------------------------- /configs/yamls/stage2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/configs/yamls/stage2.yaml -------------------------------------------------------------------------------- /configs/yamls/stage2_b.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/configs/yamls/stage2_b.yaml -------------------------------------------------------------------------------- /demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/demo.py -------------------------------------------------------------------------------- /docs/API.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/docs/API.md -------------------------------------------------------------------------------- /docs/DATASET.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/docs/DATASET.md -------------------------------------------------------------------------------- /docs/DOCKER.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/docs/DOCKER.md -------------------------------------------------------------------------------- /docs/INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/docs/INSTALL.md -------------------------------------------------------------------------------- /fetch_demo_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/fetch_demo_data.sh -------------------------------------------------------------------------------- /lib/core/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/core/loss.py -------------------------------------------------------------------------------- /lib/core/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/core/trainer.py -------------------------------------------------------------------------------- /lib/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/data/_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/data/_dataset.py -------------------------------------------------------------------------------- /lib/data/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/data/dataloader.py -------------------------------------------------------------------------------- /lib/data/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/data/datasets/__init__.py -------------------------------------------------------------------------------- /lib/data/datasets/amass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/data/datasets/amass.py -------------------------------------------------------------------------------- /lib/data/datasets/bedlam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/data/datasets/bedlam.py -------------------------------------------------------------------------------- /lib/data/datasets/dataset2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/data/datasets/dataset2d.py -------------------------------------------------------------------------------- /lib/data/datasets/dataset3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/data/datasets/dataset3d.py -------------------------------------------------------------------------------- /lib/data/datasets/dataset_custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/data/datasets/dataset_custom.py -------------------------------------------------------------------------------- /lib/data/datasets/dataset_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/data/datasets/dataset_eval.py -------------------------------------------------------------------------------- /lib/data/datasets/mixed_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/data/datasets/mixed_dataset.py -------------------------------------------------------------------------------- /lib/data/datasets/videos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/data/datasets/videos.py -------------------------------------------------------------------------------- /lib/data/utils/augmentor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/data/utils/augmentor.py -------------------------------------------------------------------------------- /lib/data/utils/normalizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/data/utils/normalizer.py -------------------------------------------------------------------------------- /lib/data_utils/amass_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/data_utils/amass_utils.py -------------------------------------------------------------------------------- /lib/data_utils/emdb_eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/data_utils/emdb_eval_utils.py -------------------------------------------------------------------------------- /lib/data_utils/rich_eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/data_utils/rich_eval_utils.py -------------------------------------------------------------------------------- /lib/data_utils/threedpw_eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/data_utils/threedpw_eval_utils.py -------------------------------------------------------------------------------- /lib/data_utils/threedpw_train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/data_utils/threedpw_train_utils.py -------------------------------------------------------------------------------- /lib/eval/eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/eval/eval_utils.py -------------------------------------------------------------------------------- /lib/eval/evaluate_3dpw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/eval/evaluate_3dpw.py -------------------------------------------------------------------------------- /lib/eval/evaluate_emdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/eval/evaluate_emdb.py -------------------------------------------------------------------------------- /lib/eval/evaluate_rich.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/eval/evaluate_rich.py -------------------------------------------------------------------------------- /lib/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/models/__init__.py -------------------------------------------------------------------------------- /lib/models/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/models/layers/__init__.py -------------------------------------------------------------------------------- /lib/models/layers/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/models/layers/modules.py -------------------------------------------------------------------------------- /lib/models/layers/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/models/layers/utils.py -------------------------------------------------------------------------------- /lib/models/preproc/backbone/hmr2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/models/preproc/backbone/hmr2.py -------------------------------------------------------------------------------- /lib/models/preproc/backbone/pose_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/models/preproc/backbone/pose_transformer.py -------------------------------------------------------------------------------- /lib/models/preproc/backbone/smpl_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/models/preproc/backbone/smpl_head.py -------------------------------------------------------------------------------- /lib/models/preproc/backbone/t_cond_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/models/preproc/backbone/t_cond_mlp.py -------------------------------------------------------------------------------- /lib/models/preproc/backbone/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/models/preproc/backbone/utils.py -------------------------------------------------------------------------------- /lib/models/preproc/backbone/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/models/preproc/backbone/vit.py -------------------------------------------------------------------------------- /lib/models/preproc/detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/models/preproc/detector.py -------------------------------------------------------------------------------- /lib/models/preproc/extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/models/preproc/extractor.py -------------------------------------------------------------------------------- /lib/models/preproc/slam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/models/preproc/slam.py -------------------------------------------------------------------------------- /lib/models/smpl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/models/smpl.py -------------------------------------------------------------------------------- /lib/models/smplify/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/models/smplify/__init__.py -------------------------------------------------------------------------------- /lib/models/smplify/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/models/smplify/losses.py -------------------------------------------------------------------------------- /lib/models/smplify/smplify.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/models/smplify/smplify.py -------------------------------------------------------------------------------- /lib/models/wham.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/models/wham.py -------------------------------------------------------------------------------- /lib/utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/utils/data_utils.py -------------------------------------------------------------------------------- /lib/utils/imutils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/utils/imutils.py -------------------------------------------------------------------------------- /lib/utils/kp_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/utils/kp_utils.py -------------------------------------------------------------------------------- /lib/utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/utils/transforms.py -------------------------------------------------------------------------------- /lib/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/utils/utils.py -------------------------------------------------------------------------------- /lib/vis/renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/vis/renderer.py -------------------------------------------------------------------------------- /lib/vis/run_vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/vis/run_vis.py -------------------------------------------------------------------------------- /lib/vis/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/lib/vis/tools.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/train.py -------------------------------------------------------------------------------- /wham_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yohanshin/WHAM/HEAD/wham_api.py --------------------------------------------------------------------------------