├── LICENSE ├── NopeSAC_Net ├── config │ ├── __init__.py │ └── config.py ├── data │ ├── __init__.py │ ├── augmentation.py │ ├── datasets │ │ ├── __init__.py │ │ ├── builtin.py │ │ └── mp3d.py │ └── planercnn_transforms.py ├── evaluation │ ├── __init__.py │ ├── detectron2coco.py │ └── mp3d_evaluation.py ├── modeling │ ├── __init__.py │ ├── camera_net │ │ ├── __init__.py │ │ ├── camera_head.py │ │ └── camera_modules.py │ ├── criterion.py │ ├── matcher.py │ ├── matching_net │ │ ├── __init__.py │ │ └── matching_head.py │ ├── meta_arch │ │ ├── __init__.py │ │ ├── camera_branch.py │ │ └── siamese_planeTR.py │ ├── planeTR_net │ │ ├── __init__.py │ │ └── planeTR_head.py │ └── transformer │ │ ├── __init__.py │ │ ├── gnn.py │ │ ├── position_encoding.py │ │ └── transformer.py ├── utils │ ├── VOCap.py │ ├── __init__.py │ ├── camera.py │ ├── mesh_utils.py │ ├── meshes.py │ ├── metrics.py │ ├── misc.py │ ├── pycococreatortools.py │ ├── textures.py │ └── vis.py └── visualization │ ├── __init__.py │ ├── vis.py │ └── visualization.py ├── README.md ├── assets └── teaser.png ├── camCls ├── kmeans_rots_32.pkl ├── kmeans_trans_32.pkl ├── scannet_kmeans_rots_32.pkl └── scannet_kmeans_trans_32.pkl ├── configs ├── Base.yaml ├── inference_mp3d.yaml ├── inference_scannet.yaml ├── train_mp3d_step1.yaml ├── train_mp3d_step2.yaml ├── train_mp3d_step3.yaml ├── train_scannet_step1.yaml └── train_scannet_step2.yaml ├── docker ├── Dockerfile └── README.md ├── environment.yaml ├── eval.py ├── test_NopeSAC.py ├── tools ├── __init__.py ├── generate_scannetv2_data_final.py └── sparseplane_planeloss.py ├── train_NopeSAC.py └── vis_NopeSAC.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/LICENSE -------------------------------------------------------------------------------- /NopeSAC_Net/config/__init__.py: -------------------------------------------------------------------------------- 1 | from .config import get_sparseplane_cfg_defaults 2 | -------------------------------------------------------------------------------- /NopeSAC_Net/config/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/config/config.py -------------------------------------------------------------------------------- /NopeSAC_Net/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/data/__init__.py -------------------------------------------------------------------------------- /NopeSAC_Net/data/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/data/augmentation.py -------------------------------------------------------------------------------- /NopeSAC_Net/data/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/data/datasets/__init__.py -------------------------------------------------------------------------------- /NopeSAC_Net/data/datasets/builtin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/data/datasets/builtin.py -------------------------------------------------------------------------------- /NopeSAC_Net/data/datasets/mp3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/data/datasets/mp3d.py -------------------------------------------------------------------------------- /NopeSAC_Net/data/planercnn_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/data/planercnn_transforms.py -------------------------------------------------------------------------------- /NopeSAC_Net/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/evaluation/__init__.py -------------------------------------------------------------------------------- /NopeSAC_Net/evaluation/detectron2coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/evaluation/detectron2coco.py -------------------------------------------------------------------------------- /NopeSAC_Net/evaluation/mp3d_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/evaluation/mp3d_evaluation.py -------------------------------------------------------------------------------- /NopeSAC_Net/modeling/__init__.py: -------------------------------------------------------------------------------- 1 | from .meta_arch import * 2 | -------------------------------------------------------------------------------- /NopeSAC_Net/modeling/camera_net/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/modeling/camera_net/__init__.py -------------------------------------------------------------------------------- /NopeSAC_Net/modeling/camera_net/camera_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/modeling/camera_net/camera_head.py -------------------------------------------------------------------------------- /NopeSAC_Net/modeling/camera_net/camera_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/modeling/camera_net/camera_modules.py -------------------------------------------------------------------------------- /NopeSAC_Net/modeling/criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/modeling/criterion.py -------------------------------------------------------------------------------- /NopeSAC_Net/modeling/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/modeling/matcher.py -------------------------------------------------------------------------------- /NopeSAC_Net/modeling/matching_net/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/modeling/matching_net/__init__.py -------------------------------------------------------------------------------- /NopeSAC_Net/modeling/matching_net/matching_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/modeling/matching_net/matching_head.py -------------------------------------------------------------------------------- /NopeSAC_Net/modeling/meta_arch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/modeling/meta_arch/__init__.py -------------------------------------------------------------------------------- /NopeSAC_Net/modeling/meta_arch/camera_branch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/modeling/meta_arch/camera_branch.py -------------------------------------------------------------------------------- /NopeSAC_Net/modeling/meta_arch/siamese_planeTR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/modeling/meta_arch/siamese_planeTR.py -------------------------------------------------------------------------------- /NopeSAC_Net/modeling/planeTR_net/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/modeling/planeTR_net/__init__.py -------------------------------------------------------------------------------- /NopeSAC_Net/modeling/planeTR_net/planeTR_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/modeling/planeTR_net/planeTR_head.py -------------------------------------------------------------------------------- /NopeSAC_Net/modeling/transformer/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. 2 | -------------------------------------------------------------------------------- /NopeSAC_Net/modeling/transformer/gnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/modeling/transformer/gnn.py -------------------------------------------------------------------------------- /NopeSAC_Net/modeling/transformer/position_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/modeling/transformer/position_encoding.py -------------------------------------------------------------------------------- /NopeSAC_Net/modeling/transformer/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/modeling/transformer/transformer.py -------------------------------------------------------------------------------- /NopeSAC_Net/utils/VOCap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/utils/VOCap.py -------------------------------------------------------------------------------- /NopeSAC_Net/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /NopeSAC_Net/utils/camera.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/utils/camera.py -------------------------------------------------------------------------------- /NopeSAC_Net/utils/mesh_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/utils/mesh_utils.py -------------------------------------------------------------------------------- /NopeSAC_Net/utils/meshes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/utils/meshes.py -------------------------------------------------------------------------------- /NopeSAC_Net/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/utils/metrics.py -------------------------------------------------------------------------------- /NopeSAC_Net/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/utils/misc.py -------------------------------------------------------------------------------- /NopeSAC_Net/utils/pycococreatortools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/utils/pycococreatortools.py -------------------------------------------------------------------------------- /NopeSAC_Net/utils/textures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/utils/textures.py -------------------------------------------------------------------------------- /NopeSAC_Net/utils/vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/utils/vis.py -------------------------------------------------------------------------------- /NopeSAC_Net/visualization/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/visualization/__init__.py -------------------------------------------------------------------------------- /NopeSAC_Net/visualization/vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/visualization/vis.py -------------------------------------------------------------------------------- /NopeSAC_Net/visualization/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/NopeSAC_Net/visualization/visualization.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/README.md -------------------------------------------------------------------------------- /assets/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/assets/teaser.png -------------------------------------------------------------------------------- /camCls/kmeans_rots_32.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/camCls/kmeans_rots_32.pkl -------------------------------------------------------------------------------- /camCls/kmeans_trans_32.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/camCls/kmeans_trans_32.pkl -------------------------------------------------------------------------------- /camCls/scannet_kmeans_rots_32.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/camCls/scannet_kmeans_rots_32.pkl -------------------------------------------------------------------------------- /camCls/scannet_kmeans_trans_32.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/camCls/scannet_kmeans_trans_32.pkl -------------------------------------------------------------------------------- /configs/Base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/configs/Base.yaml -------------------------------------------------------------------------------- /configs/inference_mp3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/configs/inference_mp3d.yaml -------------------------------------------------------------------------------- /configs/inference_scannet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/configs/inference_scannet.yaml -------------------------------------------------------------------------------- /configs/train_mp3d_step1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/configs/train_mp3d_step1.yaml -------------------------------------------------------------------------------- /configs/train_mp3d_step2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/configs/train_mp3d_step2.yaml -------------------------------------------------------------------------------- /configs/train_mp3d_step3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/configs/train_mp3d_step3.yaml -------------------------------------------------------------------------------- /configs/train_scannet_step1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/configs/train_scannet_step1.yaml -------------------------------------------------------------------------------- /configs/train_scannet_step2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/configs/train_scannet_step2.yaml -------------------------------------------------------------------------------- /docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/docker/Dockerfile -------------------------------------------------------------------------------- /docker/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/docker/README.md -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/environment.yaml -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/eval.py -------------------------------------------------------------------------------- /test_NopeSAC.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/test_NopeSAC.py -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/generate_scannetv2_data_final.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/tools/generate_scannetv2_data_final.py -------------------------------------------------------------------------------- /tools/sparseplane_planeloss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/tools/sparseplane_planeloss.py -------------------------------------------------------------------------------- /train_NopeSAC.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/train_NopeSAC.py -------------------------------------------------------------------------------- /vis_NopeSAC.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/IceTTTb/NopeSAC/HEAD/vis_NopeSAC.py --------------------------------------------------------------------------------