├── .gitignore ├── README.md ├── asset └── teaser.png ├── data ├── download_preprocess_data_here ├── generate_split_for_arkit.py └── unzip_all_data.py ├── demo.py ├── disco ├── __init__.py ├── configs │ ├── config_utils.py │ ├── finetune_triplane_diffusion.yaml │ ├── train_triplane_diffusion.yaml │ └── train_triplane_vae.yaml ├── datasets │ ├── MultiView_dataset.py │ ├── __init__.py │ ├── simple_dataset.py │ ├── taxonomy.py │ └── transforms.py ├── engine │ ├── engine_triplane_dm.py │ └── engine_triplane_vae.py ├── models │ ├── TriplaneVAE.py │ ├── Triplane_Diffusion.py │ ├── __init__.py │ └── modules │ │ ├── PointEMB.py │ │ ├── Positional_Embedding.py │ │ ├── __init__.py │ │ ├── decoder.py │ │ ├── diffusion_sampler.py │ │ ├── encoder.py │ │ ├── image_sampler.py │ │ ├── parpoints_encoder.py │ │ ├── point_transformer.py │ │ ├── pointnet2_backbone.py │ │ ├── resnet_block.py │ │ ├── resunet.py │ │ ├── unet.py │ │ └── utils.py ├── scripts │ ├── cache_img_vit_features.py │ ├── cache_triplane_vae_features.py │ ├── evaluate_reconstruction.py │ ├── train_triplane_diffusion.py │ └── train_triplane_vae.py └── utils │ ├── lr_sched.py │ ├── misc.py │ ├── projection_utils.py │ ├── pyTorchChamferDistance │ ├── .gitignore │ ├── LICENSE.md │ ├── README.md │ ├── __init__.py │ └── chamfer_distance │ │ ├── __init__.py │ │ ├── chamfer_distance.cpp │ │ ├── chamfer_distance.cu │ │ └── chamfer_distance.py │ ├── simple_image_loader.py │ └── train_test_utils.py ├── launch.py ├── output └── put_checkpoints_here ├── pyproject.toml └── requirements.txt /.gitignore: -------------------------------------------------------------------------------- 1 | *.pyc 2 | data 3 | output 4 | *.egg-info -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/README.md -------------------------------------------------------------------------------- /asset/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/asset/teaser.png -------------------------------------------------------------------------------- /data/download_preprocess_data_here: -------------------------------------------------------------------------------- 1 | 1 -------------------------------------------------------------------------------- /data/generate_split_for_arkit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/data/generate_split_for_arkit.py -------------------------------------------------------------------------------- /data/unzip_all_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/data/unzip_all_data.py -------------------------------------------------------------------------------- /demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/demo.py -------------------------------------------------------------------------------- /disco/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /disco/configs/config_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/configs/config_utils.py -------------------------------------------------------------------------------- /disco/configs/finetune_triplane_diffusion.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/configs/finetune_triplane_diffusion.yaml -------------------------------------------------------------------------------- /disco/configs/train_triplane_diffusion.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/configs/train_triplane_diffusion.yaml -------------------------------------------------------------------------------- /disco/configs/train_triplane_vae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/configs/train_triplane_vae.yaml -------------------------------------------------------------------------------- /disco/datasets/MultiView_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/datasets/MultiView_dataset.py -------------------------------------------------------------------------------- /disco/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/datasets/__init__.py -------------------------------------------------------------------------------- /disco/datasets/simple_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/datasets/simple_dataset.py -------------------------------------------------------------------------------- /disco/datasets/taxonomy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/datasets/taxonomy.py -------------------------------------------------------------------------------- /disco/datasets/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/datasets/transforms.py -------------------------------------------------------------------------------- /disco/engine/engine_triplane_dm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/engine/engine_triplane_dm.py -------------------------------------------------------------------------------- /disco/engine/engine_triplane_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/engine/engine_triplane_vae.py -------------------------------------------------------------------------------- /disco/models/TriplaneVAE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/models/TriplaneVAE.py -------------------------------------------------------------------------------- /disco/models/Triplane_Diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/models/Triplane_Diffusion.py -------------------------------------------------------------------------------- /disco/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/models/__init__.py -------------------------------------------------------------------------------- /disco/models/modules/PointEMB.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/models/modules/PointEMB.py -------------------------------------------------------------------------------- /disco/models/modules/Positional_Embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/models/modules/Positional_Embedding.py -------------------------------------------------------------------------------- /disco/models/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/models/modules/__init__.py -------------------------------------------------------------------------------- /disco/models/modules/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/models/modules/decoder.py -------------------------------------------------------------------------------- /disco/models/modules/diffusion_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/models/modules/diffusion_sampler.py -------------------------------------------------------------------------------- /disco/models/modules/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/models/modules/encoder.py -------------------------------------------------------------------------------- /disco/models/modules/image_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/models/modules/image_sampler.py -------------------------------------------------------------------------------- /disco/models/modules/parpoints_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/models/modules/parpoints_encoder.py -------------------------------------------------------------------------------- /disco/models/modules/point_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/models/modules/point_transformer.py -------------------------------------------------------------------------------- /disco/models/modules/pointnet2_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/models/modules/pointnet2_backbone.py -------------------------------------------------------------------------------- /disco/models/modules/resnet_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/models/modules/resnet_block.py -------------------------------------------------------------------------------- /disco/models/modules/resunet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/models/modules/resunet.py -------------------------------------------------------------------------------- /disco/models/modules/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/models/modules/unet.py -------------------------------------------------------------------------------- /disco/models/modules/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/models/modules/utils.py -------------------------------------------------------------------------------- /disco/scripts/cache_img_vit_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/scripts/cache_img_vit_features.py -------------------------------------------------------------------------------- /disco/scripts/cache_triplane_vae_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/scripts/cache_triplane_vae_features.py -------------------------------------------------------------------------------- /disco/scripts/evaluate_reconstruction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/scripts/evaluate_reconstruction.py -------------------------------------------------------------------------------- /disco/scripts/train_triplane_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/scripts/train_triplane_diffusion.py -------------------------------------------------------------------------------- /disco/scripts/train_triplane_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/scripts/train_triplane_vae.py -------------------------------------------------------------------------------- /disco/utils/lr_sched.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/utils/lr_sched.py -------------------------------------------------------------------------------- /disco/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/utils/misc.py -------------------------------------------------------------------------------- /disco/utils/projection_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/utils/projection_utils.py -------------------------------------------------------------------------------- /disco/utils/pyTorchChamferDistance/.gitignore: -------------------------------------------------------------------------------- 1 | .DS_Store 2 | ._* 3 | 4 | -------------------------------------------------------------------------------- /disco/utils/pyTorchChamferDistance/LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/utils/pyTorchChamferDistance/LICENSE.md -------------------------------------------------------------------------------- /disco/utils/pyTorchChamferDistance/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/utils/pyTorchChamferDistance/README.md -------------------------------------------------------------------------------- /disco/utils/pyTorchChamferDistance/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /disco/utils/pyTorchChamferDistance/chamfer_distance/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/utils/pyTorchChamferDistance/chamfer_distance/__init__.py -------------------------------------------------------------------------------- /disco/utils/pyTorchChamferDistance/chamfer_distance/chamfer_distance.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/utils/pyTorchChamferDistance/chamfer_distance/chamfer_distance.cpp -------------------------------------------------------------------------------- /disco/utils/pyTorchChamferDistance/chamfer_distance/chamfer_distance.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/utils/pyTorchChamferDistance/chamfer_distance/chamfer_distance.cu -------------------------------------------------------------------------------- /disco/utils/pyTorchChamferDistance/chamfer_distance/chamfer_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/utils/pyTorchChamferDistance/chamfer_distance/chamfer_distance.py -------------------------------------------------------------------------------- /disco/utils/simple_image_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/disco/utils/simple_image_loader.py -------------------------------------------------------------------------------- /disco/utils/train_test_utils.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /launch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/launch.py -------------------------------------------------------------------------------- /output/put_checkpoints_here: -------------------------------------------------------------------------------- 1 | 1 -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/GAP-LAB-CUHK-SZ/DisCo/HEAD/requirements.txt --------------------------------------------------------------------------------