├── .gitignore ├── LICENSE ├── README.md ├── configs ├── __init__.py ├── paths.py ├── pose2D_hrnet_config.py └── poseMF_shapeGaussian_net_config.py ├── data ├── __init__.py ├── copy_lsun_images_to_train_files_dir.py ├── on_the_fly_smpl_train_dataset.py ├── pw3d_eval_dataset.py ├── pw3d_preprocess.py └── ssp3d_eval_dataset.py ├── demo ├── 00000.png ├── 00001.png ├── 00002.png ├── 00003.png ├── 00004.png ├── 00005.png ├── 00006.png ├── 00007.png ├── 00008.png ├── 00009.png ├── 00010.png └── 00011.png ├── evaluate ├── __init__.py └── evaluate_poseMF_shapeGaussian_net.py ├── losses ├── __init__.py └── matrix_fisher_loss.py ├── metrics ├── __init__.py ├── eval_metrics_tracker.py └── train_loss_and_metrics_tracker.py ├── model_files ├── J_regressor_extra.npy ├── J_regressor_h36m.npy ├── UV_Processed.mat └── cocoplus_regressor.npy ├── models ├── __init__.py ├── canny_edge_detector.py ├── pose2D_hrnet.py ├── poseMF_shapeGaussian_net.py ├── resnet.py └── smpl_official.py ├── predict ├── __init__.py ├── predict_hrnet.py └── predict_poseMF_shapeGaussian_net.py ├── renderers ├── __init__.py └── pytorch3d_textured_renderer.py ├── requirements.txt ├── run_evaluate.py ├── run_predict.py ├── run_train.py ├── teaser.gif ├── train ├── __init__.py └── train_poseMF_shapeGaussian_net.py └── utils ├── __init__.py ├── augmentation ├── __init__.py ├── cam_augmentation.py ├── lighting_augmentation.py ├── proxy_rep_augmentation.py ├── rgb_augmentation.py └── smpl_augmentation.py ├── cam_utils.py ├── checkpoint_utils.py ├── eval_utils.py ├── image_utils.py ├── joints2d_utils.py ├── label_conversions.py ├── plot_loss_curves.py ├── rigid_transform_utils.py └── sampling_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/README.md -------------------------------------------------------------------------------- /configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/configs/paths.py -------------------------------------------------------------------------------- /configs/pose2D_hrnet_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/configs/pose2D_hrnet_config.py -------------------------------------------------------------------------------- /configs/poseMF_shapeGaussian_net_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/configs/poseMF_shapeGaussian_net_config.py -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/copy_lsun_images_to_train_files_dir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/data/copy_lsun_images_to_train_files_dir.py -------------------------------------------------------------------------------- /data/on_the_fly_smpl_train_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/data/on_the_fly_smpl_train_dataset.py -------------------------------------------------------------------------------- /data/pw3d_eval_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/data/pw3d_eval_dataset.py -------------------------------------------------------------------------------- /data/pw3d_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/data/pw3d_preprocess.py -------------------------------------------------------------------------------- /data/ssp3d_eval_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/data/ssp3d_eval_dataset.py -------------------------------------------------------------------------------- /demo/00000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/demo/00000.png -------------------------------------------------------------------------------- /demo/00001.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/demo/00001.png -------------------------------------------------------------------------------- /demo/00002.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/demo/00002.png -------------------------------------------------------------------------------- /demo/00003.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/demo/00003.png -------------------------------------------------------------------------------- /demo/00004.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/demo/00004.png -------------------------------------------------------------------------------- /demo/00005.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/demo/00005.png -------------------------------------------------------------------------------- /demo/00006.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/demo/00006.png -------------------------------------------------------------------------------- /demo/00007.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/demo/00007.png -------------------------------------------------------------------------------- /demo/00008.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/demo/00008.png -------------------------------------------------------------------------------- /demo/00009.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/demo/00009.png -------------------------------------------------------------------------------- /demo/00010.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/demo/00010.png -------------------------------------------------------------------------------- /demo/00011.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/demo/00011.png -------------------------------------------------------------------------------- /evaluate/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /evaluate/evaluate_poseMF_shapeGaussian_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/evaluate/evaluate_poseMF_shapeGaussian_net.py -------------------------------------------------------------------------------- /losses/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /losses/matrix_fisher_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/losses/matrix_fisher_loss.py -------------------------------------------------------------------------------- /metrics/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /metrics/eval_metrics_tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/metrics/eval_metrics_tracker.py -------------------------------------------------------------------------------- /metrics/train_loss_and_metrics_tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/metrics/train_loss_and_metrics_tracker.py -------------------------------------------------------------------------------- /model_files/J_regressor_extra.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/model_files/J_regressor_extra.npy -------------------------------------------------------------------------------- /model_files/J_regressor_h36m.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/model_files/J_regressor_h36m.npy -------------------------------------------------------------------------------- /model_files/UV_Processed.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/model_files/UV_Processed.mat -------------------------------------------------------------------------------- /model_files/cocoplus_regressor.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/model_files/cocoplus_regressor.npy -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/canny_edge_detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/models/canny_edge_detector.py -------------------------------------------------------------------------------- /models/pose2D_hrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/models/pose2D_hrnet.py -------------------------------------------------------------------------------- /models/poseMF_shapeGaussian_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/models/poseMF_shapeGaussian_net.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/smpl_official.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/models/smpl_official.py -------------------------------------------------------------------------------- /predict/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /predict/predict_hrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/predict/predict_hrnet.py -------------------------------------------------------------------------------- /predict/predict_poseMF_shapeGaussian_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/predict/predict_poseMF_shapeGaussian_net.py -------------------------------------------------------------------------------- /renderers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /renderers/pytorch3d_textured_renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/renderers/pytorch3d_textured_renderer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/run_evaluate.py -------------------------------------------------------------------------------- /run_predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/run_predict.py -------------------------------------------------------------------------------- /run_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/run_train.py -------------------------------------------------------------------------------- /teaser.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/teaser.gif -------------------------------------------------------------------------------- /train/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /train/train_poseMF_shapeGaussian_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/train/train_poseMF_shapeGaussian_net.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/augmentation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/augmentation/cam_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/utils/augmentation/cam_augmentation.py -------------------------------------------------------------------------------- /utils/augmentation/lighting_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/utils/augmentation/lighting_augmentation.py -------------------------------------------------------------------------------- /utils/augmentation/proxy_rep_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/utils/augmentation/proxy_rep_augmentation.py -------------------------------------------------------------------------------- /utils/augmentation/rgb_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/utils/augmentation/rgb_augmentation.py -------------------------------------------------------------------------------- /utils/augmentation/smpl_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/utils/augmentation/smpl_augmentation.py -------------------------------------------------------------------------------- /utils/cam_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/utils/cam_utils.py -------------------------------------------------------------------------------- /utils/checkpoint_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/utils/checkpoint_utils.py -------------------------------------------------------------------------------- /utils/eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/utils/eval_utils.py -------------------------------------------------------------------------------- /utils/image_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/utils/image_utils.py -------------------------------------------------------------------------------- /utils/joints2d_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/utils/joints2d_utils.py -------------------------------------------------------------------------------- /utils/label_conversions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/utils/label_conversions.py -------------------------------------------------------------------------------- /utils/plot_loss_curves.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/utils/plot_loss_curves.py -------------------------------------------------------------------------------- /utils/rigid_transform_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/utils/rigid_transform_utils.py -------------------------------------------------------------------------------- /utils/sampling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/akashsengupta1997/HierarchicalProbabilistic3DHuman/HEAD/utils/sampling_utils.py --------------------------------------------------------------------------------