├── .gitignore ├── .gitmodules ├── LICENSE ├── README.md ├── configs ├── config.yaml ├── config_all.yaml └── config_bedlam.yaml ├── data ├── README.md ├── examples │ ├── Garry_Winogrand.jpg │ ├── Neil_Leifer.jpg │ ├── multiview_examples │ │ ├── S9_Photo.54138969_001001.jpg │ │ ├── S9_Photo.55011271_001001.jpg │ │ ├── S9_Photo.58860488_001001.jpg │ │ ├── S9_Photo.60457274_001001.jpg │ │ └── multiview_examples.npz │ ├── nytimes.jpg │ └── ubc_examples │ │ ├── 0000.png │ │ ├── 0006.png │ │ ├── 0012.png │ │ ├── 0018.png │ │ ├── 0024.png │ │ ├── 0030.png │ │ ├── 0036.png │ │ ├── 0042.png │ │ ├── 0048.png │ │ ├── 0054.png │ │ ├── 0060.png │ │ ├── 0066.png │ │ ├── 0072.png │ │ ├── 0078.png │ │ ├── 0084.png │ │ ├── 0090.png │ │ ├── 0096.png │ │ ├── 0102.png │ │ ├── 0108.png │ │ ├── 0114.png │ │ ├── 0120.png │ │ └── 0126.png ├── method.png └── smpl │ ├── J_regressor_extra.npy │ ├── J_regressor_h36m.npy │ ├── smpl_dense.npy │ ├── smpl_down.npy │ ├── smpl_mean_params.npz │ └── smpl_ssm.npy ├── data_config.py ├── demo.py ├── lib ├── __init__.py ├── core │ ├── __init__.py │ ├── base_trainer.py │ ├── config.py │ ├── constants.py │ ├── data_loader.py │ └── losses.py ├── datasets │ ├── __init__.py │ ├── base_dataset.py │ ├── coco_occlusion.py │ ├── detect_dataset.py │ ├── mixed_dataset.py │ ├── mv_dataset.py │ └── preprocess │ │ ├── __init__.py │ │ ├── agora.py │ │ ├── agora2spin_bedlam.py │ │ ├── agora_projection.py │ │ ├── bedlam2spin.py │ │ ├── eft2spin.py │ │ └── pascal_occluder.py ├── get_dataloader.py ├── get_model.py ├── models │ ├── __init__.py │ ├── hrnet_refit.py │ ├── multi_linear.py │ ├── refit.py │ ├── refit_mv.py │ ├── smpl.py │ ├── update.py │ └── utils.py ├── renderer │ ├── __init__.py │ └── renderer_img.py ├── trainer.py ├── utils │ ├── __init__.py │ ├── geometry.py │ ├── imutils.py │ ├── pose_utils.py │ ├── rotation_conversions.py │ ├── utils.py │ └── visualizer.py └── yolo.py ├── requirements.txt ├── scripts ├── eval.py ├── eval_mv.py ├── multiview_demo.py ├── process_folder.py └── subsample_3dpw.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/.gitmodules -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/README.md -------------------------------------------------------------------------------- /configs/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/configs/config.yaml -------------------------------------------------------------------------------- /configs/config_all.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/configs/config_all.yaml -------------------------------------------------------------------------------- /configs/config_bedlam.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/configs/config_bedlam.yaml -------------------------------------------------------------------------------- /data/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/README.md -------------------------------------------------------------------------------- /data/examples/Garry_Winogrand.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/Garry_Winogrand.jpg -------------------------------------------------------------------------------- /data/examples/Neil_Leifer.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/Neil_Leifer.jpg -------------------------------------------------------------------------------- /data/examples/multiview_examples/S9_Photo.54138969_001001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/multiview_examples/S9_Photo.54138969_001001.jpg -------------------------------------------------------------------------------- /data/examples/multiview_examples/S9_Photo.55011271_001001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/multiview_examples/S9_Photo.55011271_001001.jpg -------------------------------------------------------------------------------- /data/examples/multiview_examples/S9_Photo.58860488_001001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/multiview_examples/S9_Photo.58860488_001001.jpg -------------------------------------------------------------------------------- /data/examples/multiview_examples/S9_Photo.60457274_001001.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/multiview_examples/S9_Photo.60457274_001001.jpg -------------------------------------------------------------------------------- /data/examples/multiview_examples/multiview_examples.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/multiview_examples/multiview_examples.npz -------------------------------------------------------------------------------- /data/examples/nytimes.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/nytimes.jpg -------------------------------------------------------------------------------- /data/examples/ubc_examples/0000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/ubc_examples/0000.png -------------------------------------------------------------------------------- /data/examples/ubc_examples/0006.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/ubc_examples/0006.png -------------------------------------------------------------------------------- /data/examples/ubc_examples/0012.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/ubc_examples/0012.png -------------------------------------------------------------------------------- /data/examples/ubc_examples/0018.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/ubc_examples/0018.png -------------------------------------------------------------------------------- /data/examples/ubc_examples/0024.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/ubc_examples/0024.png -------------------------------------------------------------------------------- /data/examples/ubc_examples/0030.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/ubc_examples/0030.png -------------------------------------------------------------------------------- /data/examples/ubc_examples/0036.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/ubc_examples/0036.png -------------------------------------------------------------------------------- /data/examples/ubc_examples/0042.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/ubc_examples/0042.png -------------------------------------------------------------------------------- /data/examples/ubc_examples/0048.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/ubc_examples/0048.png -------------------------------------------------------------------------------- /data/examples/ubc_examples/0054.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/ubc_examples/0054.png -------------------------------------------------------------------------------- /data/examples/ubc_examples/0060.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/ubc_examples/0060.png -------------------------------------------------------------------------------- /data/examples/ubc_examples/0066.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/ubc_examples/0066.png -------------------------------------------------------------------------------- /data/examples/ubc_examples/0072.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/ubc_examples/0072.png -------------------------------------------------------------------------------- /data/examples/ubc_examples/0078.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/ubc_examples/0078.png -------------------------------------------------------------------------------- /data/examples/ubc_examples/0084.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/ubc_examples/0084.png -------------------------------------------------------------------------------- /data/examples/ubc_examples/0090.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/ubc_examples/0090.png -------------------------------------------------------------------------------- /data/examples/ubc_examples/0096.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/ubc_examples/0096.png -------------------------------------------------------------------------------- /data/examples/ubc_examples/0102.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/ubc_examples/0102.png -------------------------------------------------------------------------------- /data/examples/ubc_examples/0108.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/ubc_examples/0108.png -------------------------------------------------------------------------------- /data/examples/ubc_examples/0114.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/ubc_examples/0114.png -------------------------------------------------------------------------------- /data/examples/ubc_examples/0120.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/ubc_examples/0120.png -------------------------------------------------------------------------------- /data/examples/ubc_examples/0126.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/examples/ubc_examples/0126.png -------------------------------------------------------------------------------- /data/method.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/method.png -------------------------------------------------------------------------------- /data/smpl/J_regressor_extra.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/smpl/J_regressor_extra.npy -------------------------------------------------------------------------------- /data/smpl/J_regressor_h36m.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/smpl/J_regressor_h36m.npy -------------------------------------------------------------------------------- /data/smpl/smpl_dense.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/smpl/smpl_dense.npy -------------------------------------------------------------------------------- /data/smpl/smpl_down.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/smpl/smpl_down.npy -------------------------------------------------------------------------------- /data/smpl/smpl_mean_params.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/smpl/smpl_mean_params.npz -------------------------------------------------------------------------------- /data/smpl/smpl_ssm.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data/smpl/smpl_ssm.npy -------------------------------------------------------------------------------- /data_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/data_config.py -------------------------------------------------------------------------------- /demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/demo.py -------------------------------------------------------------------------------- /lib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/__init__.py -------------------------------------------------------------------------------- /lib/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/core/__init__.py -------------------------------------------------------------------------------- /lib/core/base_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/core/base_trainer.py -------------------------------------------------------------------------------- /lib/core/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/core/config.py -------------------------------------------------------------------------------- /lib/core/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/core/constants.py -------------------------------------------------------------------------------- /lib/core/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/core/data_loader.py -------------------------------------------------------------------------------- /lib/core/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/core/losses.py -------------------------------------------------------------------------------- /lib/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/datasets/__init__.py -------------------------------------------------------------------------------- /lib/datasets/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/datasets/base_dataset.py -------------------------------------------------------------------------------- /lib/datasets/coco_occlusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/datasets/coco_occlusion.py -------------------------------------------------------------------------------- /lib/datasets/detect_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/datasets/detect_dataset.py -------------------------------------------------------------------------------- /lib/datasets/mixed_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/datasets/mixed_dataset.py -------------------------------------------------------------------------------- /lib/datasets/mv_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/datasets/mv_dataset.py -------------------------------------------------------------------------------- /lib/datasets/preprocess/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/datasets/preprocess/agora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/datasets/preprocess/agora.py -------------------------------------------------------------------------------- /lib/datasets/preprocess/agora2spin_bedlam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/datasets/preprocess/agora2spin_bedlam.py -------------------------------------------------------------------------------- /lib/datasets/preprocess/agora_projection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/datasets/preprocess/agora_projection.py -------------------------------------------------------------------------------- /lib/datasets/preprocess/bedlam2spin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/datasets/preprocess/bedlam2spin.py -------------------------------------------------------------------------------- /lib/datasets/preprocess/eft2spin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/datasets/preprocess/eft2spin.py -------------------------------------------------------------------------------- /lib/datasets/preprocess/pascal_occluder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/datasets/preprocess/pascal_occluder.py -------------------------------------------------------------------------------- /lib/get_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/get_dataloader.py -------------------------------------------------------------------------------- /lib/get_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/get_model.py -------------------------------------------------------------------------------- /lib/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/models/hrnet_refit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/models/hrnet_refit.py -------------------------------------------------------------------------------- /lib/models/multi_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/models/multi_linear.py -------------------------------------------------------------------------------- /lib/models/refit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/models/refit.py -------------------------------------------------------------------------------- /lib/models/refit_mv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/models/refit_mv.py -------------------------------------------------------------------------------- /lib/models/smpl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/models/smpl.py -------------------------------------------------------------------------------- /lib/models/update.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/models/update.py -------------------------------------------------------------------------------- /lib/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/models/utils.py -------------------------------------------------------------------------------- /lib/renderer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/renderer/renderer_img.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/renderer/renderer_img.py -------------------------------------------------------------------------------- /lib/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/trainer.py -------------------------------------------------------------------------------- /lib/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/utils/geometry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/utils/geometry.py -------------------------------------------------------------------------------- /lib/utils/imutils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/utils/imutils.py -------------------------------------------------------------------------------- /lib/utils/pose_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/utils/pose_utils.py -------------------------------------------------------------------------------- /lib/utils/rotation_conversions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/utils/rotation_conversions.py -------------------------------------------------------------------------------- /lib/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/utils/utils.py -------------------------------------------------------------------------------- /lib/utils/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/utils/visualizer.py -------------------------------------------------------------------------------- /lib/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/lib/yolo.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/scripts/eval.py -------------------------------------------------------------------------------- /scripts/eval_mv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/scripts/eval_mv.py -------------------------------------------------------------------------------- /scripts/multiview_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/scripts/multiview_demo.py -------------------------------------------------------------------------------- /scripts/process_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/scripts/process_folder.py -------------------------------------------------------------------------------- /scripts/subsample_3dpw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/scripts/subsample_3dpw.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yufu-wang/ReFit/HEAD/train.py --------------------------------------------------------------------------------