├── .gitignore ├── README.md ├── configs ├── demo.json ├── matterport3d.json ├── scannet.json └── scannetpp.json ├── dataset ├── demo.py ├── matterport.py ├── scannet.py └── scannetpp.py ├── demo.sh ├── evaluation ├── constants.py ├── evaluate.py └── utils_3d.py ├── figs └── teaser.png ├── graph ├── construction.py ├── iterative_clustering.py └── node.py ├── main.py ├── mask_predict.py ├── preprocess ├── matterport3d │ ├── category_mapping.tsv │ ├── constants.py │ └── process.py ├── scannet │ ├── SensorData.py │ ├── prepare_gt.py │ ├── process_val.py │ └── reader.py └── scannetpp │ ├── download_scannetpp.yml │ ├── prepare_iphone_data.yml │ ├── prepare_semantic_gt.yml │ ├── prepare_training_data.yml │ └── render.yml ├── pyrightconfig.json ├── requirements.txt ├── run.py ├── semantics ├── extract_label_featrues.py ├── get_open-voc_features.py └── open-voc_query.py ├── splits ├── matterport3d.txt ├── scannet.txt └── scannetpp.txt ├── utils ├── clean_all_output.py ├── config.py ├── geometry.py ├── mask_backprojection.py └── post_process.py └── visualize ├── vis_mask.py └── vis_scene.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/README.md -------------------------------------------------------------------------------- /configs/demo.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/configs/demo.json -------------------------------------------------------------------------------- /configs/matterport3d.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/configs/matterport3d.json -------------------------------------------------------------------------------- /configs/scannet.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/configs/scannet.json -------------------------------------------------------------------------------- /configs/scannetpp.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/configs/scannetpp.json -------------------------------------------------------------------------------- /dataset/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/dataset/demo.py -------------------------------------------------------------------------------- /dataset/matterport.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/dataset/matterport.py -------------------------------------------------------------------------------- /dataset/scannet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/dataset/scannet.py -------------------------------------------------------------------------------- /dataset/scannetpp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/dataset/scannetpp.py -------------------------------------------------------------------------------- /demo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/demo.sh -------------------------------------------------------------------------------- /evaluation/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/evaluation/constants.py -------------------------------------------------------------------------------- /evaluation/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/evaluation/evaluate.py -------------------------------------------------------------------------------- /evaluation/utils_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/evaluation/utils_3d.py -------------------------------------------------------------------------------- /figs/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/figs/teaser.png -------------------------------------------------------------------------------- /graph/construction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/graph/construction.py -------------------------------------------------------------------------------- /graph/iterative_clustering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/graph/iterative_clustering.py -------------------------------------------------------------------------------- /graph/node.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/graph/node.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/main.py -------------------------------------------------------------------------------- /mask_predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/mask_predict.py -------------------------------------------------------------------------------- /preprocess/matterport3d/category_mapping.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/preprocess/matterport3d/category_mapping.tsv -------------------------------------------------------------------------------- /preprocess/matterport3d/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/preprocess/matterport3d/constants.py -------------------------------------------------------------------------------- /preprocess/matterport3d/process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/preprocess/matterport3d/process.py -------------------------------------------------------------------------------- /preprocess/scannet/SensorData.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/preprocess/scannet/SensorData.py -------------------------------------------------------------------------------- /preprocess/scannet/prepare_gt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/preprocess/scannet/prepare_gt.py -------------------------------------------------------------------------------- /preprocess/scannet/process_val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/preprocess/scannet/process_val.py -------------------------------------------------------------------------------- /preprocess/scannet/reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/preprocess/scannet/reader.py -------------------------------------------------------------------------------- /preprocess/scannetpp/download_scannetpp.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/preprocess/scannetpp/download_scannetpp.yml -------------------------------------------------------------------------------- /preprocess/scannetpp/prepare_iphone_data.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/preprocess/scannetpp/prepare_iphone_data.yml -------------------------------------------------------------------------------- /preprocess/scannetpp/prepare_semantic_gt.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/preprocess/scannetpp/prepare_semantic_gt.yml -------------------------------------------------------------------------------- /preprocess/scannetpp/prepare_training_data.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/preprocess/scannetpp/prepare_training_data.yml -------------------------------------------------------------------------------- /preprocess/scannetpp/render.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/preprocess/scannetpp/render.yml -------------------------------------------------------------------------------- /pyrightconfig.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/pyrightconfig.json -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/run.py -------------------------------------------------------------------------------- /semantics/extract_label_featrues.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/semantics/extract_label_featrues.py -------------------------------------------------------------------------------- /semantics/get_open-voc_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/semantics/get_open-voc_features.py -------------------------------------------------------------------------------- /semantics/open-voc_query.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/semantics/open-voc_query.py -------------------------------------------------------------------------------- /splits/matterport3d.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/splits/matterport3d.txt -------------------------------------------------------------------------------- /splits/scannet.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/splits/scannet.txt -------------------------------------------------------------------------------- /splits/scannetpp.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/splits/scannetpp.txt -------------------------------------------------------------------------------- /utils/clean_all_output.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/utils/clean_all_output.py -------------------------------------------------------------------------------- /utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/utils/config.py -------------------------------------------------------------------------------- /utils/geometry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/utils/geometry.py -------------------------------------------------------------------------------- /utils/mask_backprojection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/utils/mask_backprojection.py -------------------------------------------------------------------------------- /utils/post_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/utils/post_process.py -------------------------------------------------------------------------------- /visualize/vis_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/visualize/vis_mask.py -------------------------------------------------------------------------------- /visualize/vis_scene.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PKU-EPIC/MaskClustering/HEAD/visualize/vis_scene.py --------------------------------------------------------------------------------