├── .dvc ├── .gitignore └── config ├── .flake8 ├── .gitignore ├── .pre-commit-config.yaml ├── README.md ├── SpatioTemporalSegmentation-ScanNet.patch ├── data ├── processed │ └── .gitkeep └── raw │ └── .gitkeep ├── docs ├── 3dv_poster.pdf ├── github_teaser.jpg ├── index.html ├── mix3d.pdf ├── paper.jpg ├── preview_poster.png ├── scannet_benchmark_screenshot.png ├── teaser.jpg └── w3.css ├── dvc.lock ├── dvc.yaml ├── kpconv_pytorch_mix3d.patch ├── kpconv_tensorflow_mix3d.patch ├── mix3d ├── __init__.py ├── __main__.py ├── conf │ ├── __init__.py │ ├── augmentation │ │ ├── albumentations_aug.yaml │ │ └── volumentations_aug.yaml │ ├── callbacks │ │ └── callbacks.yaml │ ├── config.yaml │ ├── data │ │ ├── collation_functions │ │ │ ├── voxelize_collate.yaml │ │ │ └── voxelize_collate_merge.yaml │ │ ├── data_loaders │ │ │ └── simple_loader.yaml │ │ ├── datasets │ │ │ ├── matterport.yaml │ │ │ ├── matterport_scannet.yaml │ │ │ ├── rio.yaml │ │ │ ├── s3dis.yaml │ │ │ ├── scannet.yaml │ │ │ └── semantic_kitti.yaml │ │ ├── indoor.yaml │ │ └── outdoor.yaml │ ├── logging │ │ ├── base.yaml │ │ ├── minimal.yaml │ │ └── offline.yaml │ ├── loss │ │ └── cross_entropy.yaml │ ├── metrics │ │ └── miou.yaml │ ├── model │ │ ├── baseline.yaml │ │ └── outdoor_small.yaml │ ├── optimizer │ │ ├── adamw.yaml │ │ └── adamw_lower.yaml │ ├── scheduler │ │ └── onecyclelr.yaml │ └── trainer │ │ ├── short_trainer.yaml │ │ └── trainer.yaml ├── datasets │ ├── outdoor_semseg.py │ ├── preprocessing │ │ ├── base_preprocessing.py │ │ ├── matterport_preprocessing.py │ │ ├── rio_preprocessing.py │ │ ├── scannet_preprocessing.py │ │ └── semantic_kitti_preprocessing.py │ ├── semseg.py │ └── utils.py ├── logger │ ├── __init__.py │ ├── csv_logger.py │ ├── json_logger.py │ └── logger_utils.py ├── models │ ├── __init__.py │ ├── conditional_random_fields.py │ ├── metrics │ │ ├── __init__.py │ │ ├── confusionmatrix.py │ │ └── metrics.py │ ├── model.py │ ├── modules │ │ ├── __init__.py │ │ ├── common.py │ │ ├── resnet_block.py │ │ └── senet_block.py │ ├── res16unet.py │ ├── resnet.py │ ├── resunet.py │ └── wrapper.py ├── trainer │ ├── __init__.py │ └── trainer.py └── utils │ ├── __init__.py │ ├── gradflow_check.py │ ├── kfold.py │ ├── pc_visualizations.py │ ├── point_cloud_utils.py │ └── utils.py ├── poetry.lock ├── pyproject.toml └── scripts ├── experiments └── 1000_scene_merging.bash ├── init.bash ├── local_run.bash ├── preprocess_matterport.bash ├── preprocess_rio.bash ├── preprocess_scannet.bash └── preprocess_semantic_kitti.bash /.dvc/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/.dvc/.gitignore -------------------------------------------------------------------------------- /.dvc/config: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/.flake8 -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/README.md -------------------------------------------------------------------------------- /SpatioTemporalSegmentation-ScanNet.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/SpatioTemporalSegmentation-ScanNet.patch -------------------------------------------------------------------------------- /data/processed/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/raw/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /docs/3dv_poster.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/docs/3dv_poster.pdf -------------------------------------------------------------------------------- /docs/github_teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/docs/github_teaser.jpg -------------------------------------------------------------------------------- /docs/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/docs/index.html -------------------------------------------------------------------------------- /docs/mix3d.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/docs/mix3d.pdf -------------------------------------------------------------------------------- /docs/paper.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/docs/paper.jpg -------------------------------------------------------------------------------- /docs/preview_poster.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/docs/preview_poster.png -------------------------------------------------------------------------------- /docs/scannet_benchmark_screenshot.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/docs/scannet_benchmark_screenshot.png -------------------------------------------------------------------------------- /docs/teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/docs/teaser.jpg -------------------------------------------------------------------------------- /docs/w3.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/docs/w3.css -------------------------------------------------------------------------------- /dvc.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/dvc.lock -------------------------------------------------------------------------------- /dvc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/dvc.yaml -------------------------------------------------------------------------------- /kpconv_pytorch_mix3d.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/kpconv_pytorch_mix3d.patch -------------------------------------------------------------------------------- /kpconv_tensorflow_mix3d.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/kpconv_tensorflow_mix3d.patch -------------------------------------------------------------------------------- /mix3d/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/__init__.py -------------------------------------------------------------------------------- /mix3d/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/__main__.py -------------------------------------------------------------------------------- /mix3d/conf/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mix3d/conf/augmentation/albumentations_aug.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/conf/augmentation/albumentations_aug.yaml -------------------------------------------------------------------------------- /mix3d/conf/augmentation/volumentations_aug.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/conf/augmentation/volumentations_aug.yaml -------------------------------------------------------------------------------- /mix3d/conf/callbacks/callbacks.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/conf/callbacks/callbacks.yaml -------------------------------------------------------------------------------- /mix3d/conf/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/conf/config.yaml -------------------------------------------------------------------------------- /mix3d/conf/data/collation_functions/voxelize_collate.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/conf/data/collation_functions/voxelize_collate.yaml -------------------------------------------------------------------------------- /mix3d/conf/data/collation_functions/voxelize_collate_merge.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/conf/data/collation_functions/voxelize_collate_merge.yaml -------------------------------------------------------------------------------- /mix3d/conf/data/data_loaders/simple_loader.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/conf/data/data_loaders/simple_loader.yaml -------------------------------------------------------------------------------- /mix3d/conf/data/datasets/matterport.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/conf/data/datasets/matterport.yaml -------------------------------------------------------------------------------- /mix3d/conf/data/datasets/matterport_scannet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/conf/data/datasets/matterport_scannet.yaml -------------------------------------------------------------------------------- /mix3d/conf/data/datasets/rio.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/conf/data/datasets/rio.yaml -------------------------------------------------------------------------------- /mix3d/conf/data/datasets/s3dis.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/conf/data/datasets/s3dis.yaml -------------------------------------------------------------------------------- /mix3d/conf/data/datasets/scannet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/conf/data/datasets/scannet.yaml -------------------------------------------------------------------------------- /mix3d/conf/data/datasets/semantic_kitti.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/conf/data/datasets/semantic_kitti.yaml -------------------------------------------------------------------------------- /mix3d/conf/data/indoor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/conf/data/indoor.yaml -------------------------------------------------------------------------------- /mix3d/conf/data/outdoor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/conf/data/outdoor.yaml -------------------------------------------------------------------------------- /mix3d/conf/logging/base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/conf/logging/base.yaml -------------------------------------------------------------------------------- /mix3d/conf/logging/minimal.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/conf/logging/minimal.yaml -------------------------------------------------------------------------------- /mix3d/conf/logging/offline.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/conf/logging/offline.yaml -------------------------------------------------------------------------------- /mix3d/conf/loss/cross_entropy.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/conf/loss/cross_entropy.yaml -------------------------------------------------------------------------------- /mix3d/conf/metrics/miou.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/conf/metrics/miou.yaml -------------------------------------------------------------------------------- /mix3d/conf/model/baseline.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/conf/model/baseline.yaml -------------------------------------------------------------------------------- /mix3d/conf/model/outdoor_small.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/conf/model/outdoor_small.yaml -------------------------------------------------------------------------------- /mix3d/conf/optimizer/adamw.yaml: -------------------------------------------------------------------------------- 1 | # @package _group_ 2 | _target_: torch.optim.AdamW 3 | lr: 0.01 4 | -------------------------------------------------------------------------------- /mix3d/conf/optimizer/adamw_lower.yaml: -------------------------------------------------------------------------------- 1 | # @package _group_ 2 | _target_: torch.optim.AdamW 3 | lr: 0.005 4 | -------------------------------------------------------------------------------- /mix3d/conf/scheduler/onecyclelr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/conf/scheduler/onecyclelr.yaml -------------------------------------------------------------------------------- /mix3d/conf/trainer/short_trainer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/conf/trainer/short_trainer.yaml -------------------------------------------------------------------------------- /mix3d/conf/trainer/trainer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/conf/trainer/trainer.yaml -------------------------------------------------------------------------------- /mix3d/datasets/outdoor_semseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/datasets/outdoor_semseg.py -------------------------------------------------------------------------------- /mix3d/datasets/preprocessing/base_preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/datasets/preprocessing/base_preprocessing.py -------------------------------------------------------------------------------- /mix3d/datasets/preprocessing/matterport_preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/datasets/preprocessing/matterport_preprocessing.py -------------------------------------------------------------------------------- /mix3d/datasets/preprocessing/rio_preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/datasets/preprocessing/rio_preprocessing.py -------------------------------------------------------------------------------- /mix3d/datasets/preprocessing/scannet_preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/datasets/preprocessing/scannet_preprocessing.py -------------------------------------------------------------------------------- /mix3d/datasets/preprocessing/semantic_kitti_preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/datasets/preprocessing/semantic_kitti_preprocessing.py -------------------------------------------------------------------------------- /mix3d/datasets/semseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/datasets/semseg.py -------------------------------------------------------------------------------- /mix3d/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/datasets/utils.py -------------------------------------------------------------------------------- /mix3d/logger/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mix3d/logger/csv_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/logger/csv_logger.py -------------------------------------------------------------------------------- /mix3d/logger/json_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/logger/json_logger.py -------------------------------------------------------------------------------- /mix3d/logger/logger_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/logger/logger_utils.py -------------------------------------------------------------------------------- /mix3d/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/models/__init__.py -------------------------------------------------------------------------------- /mix3d/models/conditional_random_fields.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/models/conditional_random_fields.py -------------------------------------------------------------------------------- /mix3d/models/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/models/metrics/__init__.py -------------------------------------------------------------------------------- /mix3d/models/metrics/confusionmatrix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/models/metrics/confusionmatrix.py -------------------------------------------------------------------------------- /mix3d/models/metrics/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/models/metrics/metrics.py -------------------------------------------------------------------------------- /mix3d/models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/models/model.py -------------------------------------------------------------------------------- /mix3d/models/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mix3d/models/modules/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/models/modules/common.py -------------------------------------------------------------------------------- /mix3d/models/modules/resnet_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/models/modules/resnet_block.py -------------------------------------------------------------------------------- /mix3d/models/modules/senet_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/models/modules/senet_block.py -------------------------------------------------------------------------------- /mix3d/models/res16unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/models/res16unet.py -------------------------------------------------------------------------------- /mix3d/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/models/resnet.py -------------------------------------------------------------------------------- /mix3d/models/resunet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/models/resunet.py -------------------------------------------------------------------------------- /mix3d/models/wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/models/wrapper.py -------------------------------------------------------------------------------- /mix3d/trainer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mix3d/trainer/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/trainer/trainer.py -------------------------------------------------------------------------------- /mix3d/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /mix3d/utils/gradflow_check.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/utils/gradflow_check.py -------------------------------------------------------------------------------- /mix3d/utils/kfold.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/utils/kfold.py -------------------------------------------------------------------------------- /mix3d/utils/pc_visualizations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/utils/pc_visualizations.py -------------------------------------------------------------------------------- /mix3d/utils/point_cloud_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/utils/point_cloud_utils.py -------------------------------------------------------------------------------- /mix3d/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/mix3d/utils/utils.py -------------------------------------------------------------------------------- /poetry.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/poetry.lock -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/pyproject.toml -------------------------------------------------------------------------------- /scripts/experiments/1000_scene_merging.bash: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/scripts/experiments/1000_scene_merging.bash -------------------------------------------------------------------------------- /scripts/init.bash: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/scripts/init.bash -------------------------------------------------------------------------------- /scripts/local_run.bash: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/scripts/local_run.bash -------------------------------------------------------------------------------- /scripts/preprocess_matterport.bash: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/scripts/preprocess_matterport.bash -------------------------------------------------------------------------------- /scripts/preprocess_rio.bash: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/scripts/preprocess_rio.bash -------------------------------------------------------------------------------- /scripts/preprocess_scannet.bash: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/scripts/preprocess_scannet.bash -------------------------------------------------------------------------------- /scripts/preprocess_semantic_kitti.bash: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kumuji/mix3d/HEAD/scripts/preprocess_semantic_kitti.bash --------------------------------------------------------------------------------