├── .gitignore ├── README.md ├── config_loader.py ├── configs ├── arkitscenes.txt ├── s3dis_detections_learnedPS_voxsem_fold5.txt ├── s3dis_fold1.txt ├── s3dis_fold2.txt ├── s3dis_fold3.txt ├── s3dis_fold4.txt ├── s3dis_fold5.txt ├── s3dis_fold6.txt ├── scannet.txt ├── scannet_dropout1.txt ├── scannet_dropout10.txt ├── scannet_dropout2.txt ├── scannet_dropout20.txt ├── scannet_dropout5.txt ├── scannet_noisy1.txt ├── scannet_noisy10.txt ├── scannet_noisy2.txt └── scannet_noisy5.txt ├── data ├── .gitkeep ├── augmented_BBs │ ├── README.md │ └── visualize_bbs_data.py └── scannet │ └── scannetv2_official_split.npz ├── dataprocessing ├── arkitscenes.py ├── augmentation.py ├── mix3d_albumentations_aug.yaml ├── oversegmentation │ ├── README.md │ ├── cpp │ │ ├── CMakeLists.txt │ │ ├── Makefile │ │ ├── segmentator.cpp │ │ ├── tiny_obj_loader.h │ │ ├── tinyply.cpp │ │ └── tinyply.h │ ├── run_segmentator.py │ ├── scene0776_00_oversegmentation.png │ └── visualize_segments.py ├── prepare_s3dis.py ├── s3dis.py └── scannet.py ├── docs ├── arkitscenes.md ├── code_structure.md ├── installation.md └── s3dis.md ├── env.yml ├── models ├── dataloader.py ├── detection_net.py ├── evaluation.py ├── iou_nms.py ├── model.py ├── resnet.py └── training.py ├── teaser.jpeg └── utils ├── __init__.py ├── box_util.py ├── eval_metric.py ├── evaluate_detections.py ├── gt2eval.py ├── metric_util.py ├── s3dis_util.py └── util.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/README.md -------------------------------------------------------------------------------- /config_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/config_loader.py -------------------------------------------------------------------------------- /configs/arkitscenes.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/configs/arkitscenes.txt -------------------------------------------------------------------------------- /configs/s3dis_detections_learnedPS_voxsem_fold5.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/configs/s3dis_detections_learnedPS_voxsem_fold5.txt -------------------------------------------------------------------------------- /configs/s3dis_fold1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/configs/s3dis_fold1.txt -------------------------------------------------------------------------------- /configs/s3dis_fold2.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/configs/s3dis_fold2.txt -------------------------------------------------------------------------------- /configs/s3dis_fold3.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/configs/s3dis_fold3.txt -------------------------------------------------------------------------------- /configs/s3dis_fold4.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/configs/s3dis_fold4.txt -------------------------------------------------------------------------------- /configs/s3dis_fold5.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/configs/s3dis_fold5.txt -------------------------------------------------------------------------------- /configs/s3dis_fold6.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/configs/s3dis_fold6.txt -------------------------------------------------------------------------------- /configs/scannet.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/configs/scannet.txt -------------------------------------------------------------------------------- /configs/scannet_dropout1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/configs/scannet_dropout1.txt -------------------------------------------------------------------------------- /configs/scannet_dropout10.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/configs/scannet_dropout10.txt -------------------------------------------------------------------------------- /configs/scannet_dropout2.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/configs/scannet_dropout2.txt -------------------------------------------------------------------------------- /configs/scannet_dropout20.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/configs/scannet_dropout20.txt -------------------------------------------------------------------------------- /configs/scannet_dropout5.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/configs/scannet_dropout5.txt -------------------------------------------------------------------------------- /configs/scannet_noisy1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/configs/scannet_noisy1.txt -------------------------------------------------------------------------------- /configs/scannet_noisy10.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/configs/scannet_noisy10.txt -------------------------------------------------------------------------------- /configs/scannet_noisy2.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/configs/scannet_noisy2.txt -------------------------------------------------------------------------------- /configs/scannet_noisy5.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/configs/scannet_noisy5.txt -------------------------------------------------------------------------------- /data/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/augmented_BBs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/data/augmented_BBs/README.md -------------------------------------------------------------------------------- /data/augmented_BBs/visualize_bbs_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/data/augmented_BBs/visualize_bbs_data.py -------------------------------------------------------------------------------- /data/scannet/scannetv2_official_split.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/data/scannet/scannetv2_official_split.npz -------------------------------------------------------------------------------- /dataprocessing/arkitscenes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/dataprocessing/arkitscenes.py -------------------------------------------------------------------------------- /dataprocessing/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/dataprocessing/augmentation.py -------------------------------------------------------------------------------- /dataprocessing/mix3d_albumentations_aug.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/dataprocessing/mix3d_albumentations_aug.yaml -------------------------------------------------------------------------------- /dataprocessing/oversegmentation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/dataprocessing/oversegmentation/README.md -------------------------------------------------------------------------------- /dataprocessing/oversegmentation/cpp/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/dataprocessing/oversegmentation/cpp/CMakeLists.txt -------------------------------------------------------------------------------- /dataprocessing/oversegmentation/cpp/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/dataprocessing/oversegmentation/cpp/Makefile -------------------------------------------------------------------------------- /dataprocessing/oversegmentation/cpp/segmentator.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/dataprocessing/oversegmentation/cpp/segmentator.cpp -------------------------------------------------------------------------------- /dataprocessing/oversegmentation/cpp/tiny_obj_loader.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/dataprocessing/oversegmentation/cpp/tiny_obj_loader.h -------------------------------------------------------------------------------- /dataprocessing/oversegmentation/cpp/tinyply.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/dataprocessing/oversegmentation/cpp/tinyply.cpp -------------------------------------------------------------------------------- /dataprocessing/oversegmentation/cpp/tinyply.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/dataprocessing/oversegmentation/cpp/tinyply.h -------------------------------------------------------------------------------- /dataprocessing/oversegmentation/run_segmentator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/dataprocessing/oversegmentation/run_segmentator.py -------------------------------------------------------------------------------- /dataprocessing/oversegmentation/scene0776_00_oversegmentation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/dataprocessing/oversegmentation/scene0776_00_oversegmentation.png -------------------------------------------------------------------------------- /dataprocessing/oversegmentation/visualize_segments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/dataprocessing/oversegmentation/visualize_segments.py -------------------------------------------------------------------------------- /dataprocessing/prepare_s3dis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/dataprocessing/prepare_s3dis.py -------------------------------------------------------------------------------- /dataprocessing/s3dis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/dataprocessing/s3dis.py -------------------------------------------------------------------------------- /dataprocessing/scannet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/dataprocessing/scannet.py -------------------------------------------------------------------------------- /docs/arkitscenes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/docs/arkitscenes.md -------------------------------------------------------------------------------- /docs/code_structure.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/docs/code_structure.md -------------------------------------------------------------------------------- /docs/installation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/docs/installation.md -------------------------------------------------------------------------------- /docs/s3dis.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/docs/s3dis.md -------------------------------------------------------------------------------- /env.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/env.yml -------------------------------------------------------------------------------- /models/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/models/dataloader.py -------------------------------------------------------------------------------- /models/detection_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/models/detection_net.py -------------------------------------------------------------------------------- /models/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/models/evaluation.py -------------------------------------------------------------------------------- /models/iou_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/models/iou_nms.py -------------------------------------------------------------------------------- /models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/models/model.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/models/training.py -------------------------------------------------------------------------------- /teaser.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/teaser.jpeg -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/box_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/utils/box_util.py -------------------------------------------------------------------------------- /utils/eval_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/utils/eval_metric.py -------------------------------------------------------------------------------- /utils/evaluate_detections.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/utils/evaluate_detections.py -------------------------------------------------------------------------------- /utils/gt2eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/utils/gt2eval.py -------------------------------------------------------------------------------- /utils/metric_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/utils/metric_util.py -------------------------------------------------------------------------------- /utils/s3dis_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/utils/s3dis_util.py -------------------------------------------------------------------------------- /utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jchibane/Box2Mask/HEAD/utils/util.py --------------------------------------------------------------------------------