├── .gitignore ├── NTUitive Dual License.rtf ├── README.md ├── anyloc ├── dino_extractor.py └── utilities.py ├── commons.py ├── compute_other_loss.py ├── datasets ├── datasets_ws_kitti360.py └── datasets_ws_nuscenes.py ├── datasets_ws.py ├── layers ├── eca_block.py └── pooling.py ├── model ├── __init__.py ├── aggregation.py ├── cct │ ├── __init__.py │ ├── cct.py │ ├── embedder.py │ ├── helpers.py │ ├── stochastic_depth.py │ ├── tokenizer.py │ └── transformers.py ├── functional.py ├── network.py ├── normalization.py └── sync_batchnorm │ ├── __init__.py │ ├── batchnorm.py │ ├── batchnorm_reimpl.py │ ├── comm.py │ ├── replicate.py │ └── unittest.py ├── models ├── loss.py ├── minkfpn.py ├── minkloc.py ├── minkloc3d.txt ├── minkloc_multimodal.py ├── minklocmultimodal.txt ├── minklocrgb.txt ├── model_factory.py └── resnet.py ├── models_baseline └── dbvanilla2d.py ├── models_minkloc ├── eca_block.py ├── loss.py ├── minkfpn.py ├── minkloc.py ├── pooling.py └── resnet.py ├── network ├── image_fe.py └── image_pooling.py ├── network_mm ├── diff_block.py ├── ffns.py ├── fuse_block_toshallow.py ├── gnns.py ├── image_fe.py ├── image_pooling.py ├── mm.py └── stage2fuse_blockadd.py ├── pc_augmentation.py ├── script.sh ├── teaser.png ├── test.py ├── tools └── options.py ├── train.py ├── util.py └── viz_lidar.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/.gitignore -------------------------------------------------------------------------------- /NTUitive Dual License.rtf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/NTUitive Dual License.rtf -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/README.md -------------------------------------------------------------------------------- /anyloc/dino_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/anyloc/dino_extractor.py -------------------------------------------------------------------------------- /anyloc/utilities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/anyloc/utilities.py -------------------------------------------------------------------------------- /commons.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/commons.py -------------------------------------------------------------------------------- /compute_other_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/compute_other_loss.py -------------------------------------------------------------------------------- /datasets/datasets_ws_kitti360.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/datasets/datasets_ws_kitti360.py -------------------------------------------------------------------------------- /datasets/datasets_ws_nuscenes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/datasets/datasets_ws_nuscenes.py -------------------------------------------------------------------------------- /datasets_ws.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/datasets_ws.py -------------------------------------------------------------------------------- /layers/eca_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/layers/eca_block.py -------------------------------------------------------------------------------- /layers/pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/layers/pooling.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/aggregation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/model/aggregation.py -------------------------------------------------------------------------------- /model/cct/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/model/cct/__init__.py -------------------------------------------------------------------------------- /model/cct/cct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/model/cct/cct.py -------------------------------------------------------------------------------- /model/cct/embedder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/model/cct/embedder.py -------------------------------------------------------------------------------- /model/cct/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/model/cct/helpers.py -------------------------------------------------------------------------------- /model/cct/stochastic_depth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/model/cct/stochastic_depth.py -------------------------------------------------------------------------------- /model/cct/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/model/cct/tokenizer.py -------------------------------------------------------------------------------- /model/cct/transformers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/model/cct/transformers.py -------------------------------------------------------------------------------- /model/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/model/functional.py -------------------------------------------------------------------------------- /model/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/model/network.py -------------------------------------------------------------------------------- /model/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/model/normalization.py -------------------------------------------------------------------------------- /model/sync_batchnorm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/model/sync_batchnorm/__init__.py -------------------------------------------------------------------------------- /model/sync_batchnorm/batchnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/model/sync_batchnorm/batchnorm.py -------------------------------------------------------------------------------- /model/sync_batchnorm/batchnorm_reimpl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/model/sync_batchnorm/batchnorm_reimpl.py -------------------------------------------------------------------------------- /model/sync_batchnorm/comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/model/sync_batchnorm/comm.py -------------------------------------------------------------------------------- /model/sync_batchnorm/replicate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/model/sync_batchnorm/replicate.py -------------------------------------------------------------------------------- /model/sync_batchnorm/unittest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/model/sync_batchnorm/unittest.py -------------------------------------------------------------------------------- /models/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/models/loss.py -------------------------------------------------------------------------------- /models/minkfpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/models/minkfpn.py -------------------------------------------------------------------------------- /models/minkloc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/models/minkloc.py -------------------------------------------------------------------------------- /models/minkloc3d.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/models/minkloc3d.txt -------------------------------------------------------------------------------- /models/minkloc_multimodal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/models/minkloc_multimodal.py -------------------------------------------------------------------------------- /models/minklocmultimodal.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/models/minklocmultimodal.txt -------------------------------------------------------------------------------- /models/minklocrgb.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/models/minklocrgb.txt -------------------------------------------------------------------------------- /models/model_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/models/model_factory.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models_baseline/dbvanilla2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/models_baseline/dbvanilla2d.py -------------------------------------------------------------------------------- /models_minkloc/eca_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/models_minkloc/eca_block.py -------------------------------------------------------------------------------- /models_minkloc/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/models_minkloc/loss.py -------------------------------------------------------------------------------- /models_minkloc/minkfpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/models_minkloc/minkfpn.py -------------------------------------------------------------------------------- /models_minkloc/minkloc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/models_minkloc/minkloc.py -------------------------------------------------------------------------------- /models_minkloc/pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/models_minkloc/pooling.py -------------------------------------------------------------------------------- /models_minkloc/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/models_minkloc/resnet.py -------------------------------------------------------------------------------- /network/image_fe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/network/image_fe.py -------------------------------------------------------------------------------- /network/image_pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/network/image_pooling.py -------------------------------------------------------------------------------- /network_mm/diff_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/network_mm/diff_block.py -------------------------------------------------------------------------------- /network_mm/ffns.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/network_mm/ffns.py -------------------------------------------------------------------------------- /network_mm/fuse_block_toshallow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/network_mm/fuse_block_toshallow.py -------------------------------------------------------------------------------- /network_mm/gnns.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/network_mm/gnns.py -------------------------------------------------------------------------------- /network_mm/image_fe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/network_mm/image_fe.py -------------------------------------------------------------------------------- /network_mm/image_pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/network_mm/image_pooling.py -------------------------------------------------------------------------------- /network_mm/mm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/network_mm/mm.py -------------------------------------------------------------------------------- /network_mm/stage2fuse_blockadd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/network_mm/stage2fuse_blockadd.py -------------------------------------------------------------------------------- /pc_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/pc_augmentation.py -------------------------------------------------------------------------------- /script.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/script.sh -------------------------------------------------------------------------------- /teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/teaser.png -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/test.py -------------------------------------------------------------------------------- /tools/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/tools/options.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/train.py -------------------------------------------------------------------------------- /util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/util.py -------------------------------------------------------------------------------- /viz_lidar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sijieaaa/AGPlace/HEAD/viz_lidar.py --------------------------------------------------------------------------------