├── .gitignore ├── .idea └── .gitignore ├── LICENSE ├── README.md ├── config └── config_egonn.txt ├── datasets ├── augmentation.py ├── base_datasets.py ├── dataset_utils.py ├── kitti │ ├── generate_evaluation_sets.py │ ├── kitti_raw.py │ └── utils.py ├── mulran │ ├── generate_evaluation_sets.py │ ├── generate_training_tuples.py │ ├── mulran_raw.py │ ├── mulran_train.py │ └── utils.py ├── quantization.py ├── samplers.py └── southbay │ ├── generate_evaluation_sets.py │ ├── generate_training_tuples.py │ └── southbay_raw.py ├── eval ├── evaluate.py └── evaluate_with_rotations.py ├── images ├── key1_.png ├── keypoints_vis.png ├── pair2.png └── registered_pairs.png ├── layers ├── eca_block.py ├── netvlad.py ├── pooling.py └── senet_block.py ├── misc ├── point_clouds.py ├── poses.py └── utils.py ├── models ├── egonn.txt ├── loss.py ├── loss_utils.py ├── minkfpn.py ├── minkgl.py ├── minkloc.py ├── minkloc3d_mulran.txt ├── model_factory.py └── resnet.py ├── third_party ├── minkloc3d │ └── minkloc.py ├── pypcd.py └── scan_context │ ├── evaluate_scan_context.py │ └── scan_context.py ├── training ├── train.py └── trainer.py └── weights └── model_egonn_20210916_1104.pth /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/.gitignore -------------------------------------------------------------------------------- /.idea/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/.idea/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/README.md -------------------------------------------------------------------------------- /config/config_egonn.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/config/config_egonn.txt -------------------------------------------------------------------------------- /datasets/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/datasets/augmentation.py -------------------------------------------------------------------------------- /datasets/base_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/datasets/base_datasets.py -------------------------------------------------------------------------------- /datasets/dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/datasets/dataset_utils.py -------------------------------------------------------------------------------- /datasets/kitti/generate_evaluation_sets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/datasets/kitti/generate_evaluation_sets.py -------------------------------------------------------------------------------- /datasets/kitti/kitti_raw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/datasets/kitti/kitti_raw.py -------------------------------------------------------------------------------- /datasets/kitti/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/datasets/kitti/utils.py -------------------------------------------------------------------------------- /datasets/mulran/generate_evaluation_sets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/datasets/mulran/generate_evaluation_sets.py -------------------------------------------------------------------------------- /datasets/mulran/generate_training_tuples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/datasets/mulran/generate_training_tuples.py -------------------------------------------------------------------------------- /datasets/mulran/mulran_raw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/datasets/mulran/mulran_raw.py -------------------------------------------------------------------------------- /datasets/mulran/mulran_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/datasets/mulran/mulran_train.py -------------------------------------------------------------------------------- /datasets/mulran/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/datasets/mulran/utils.py -------------------------------------------------------------------------------- /datasets/quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/datasets/quantization.py -------------------------------------------------------------------------------- /datasets/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/datasets/samplers.py -------------------------------------------------------------------------------- /datasets/southbay/generate_evaluation_sets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/datasets/southbay/generate_evaluation_sets.py -------------------------------------------------------------------------------- /datasets/southbay/generate_training_tuples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/datasets/southbay/generate_training_tuples.py -------------------------------------------------------------------------------- /datasets/southbay/southbay_raw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/datasets/southbay/southbay_raw.py -------------------------------------------------------------------------------- /eval/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/eval/evaluate.py -------------------------------------------------------------------------------- /eval/evaluate_with_rotations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/eval/evaluate_with_rotations.py -------------------------------------------------------------------------------- /images/key1_.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/images/key1_.png -------------------------------------------------------------------------------- /images/keypoints_vis.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/images/keypoints_vis.png -------------------------------------------------------------------------------- /images/pair2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/images/pair2.png -------------------------------------------------------------------------------- /images/registered_pairs.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/images/registered_pairs.png -------------------------------------------------------------------------------- /layers/eca_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/layers/eca_block.py -------------------------------------------------------------------------------- /layers/netvlad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/layers/netvlad.py -------------------------------------------------------------------------------- /layers/pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/layers/pooling.py -------------------------------------------------------------------------------- /layers/senet_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/layers/senet_block.py -------------------------------------------------------------------------------- /misc/point_clouds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/misc/point_clouds.py -------------------------------------------------------------------------------- /misc/poses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/misc/poses.py -------------------------------------------------------------------------------- /misc/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/misc/utils.py -------------------------------------------------------------------------------- /models/egonn.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/models/egonn.txt -------------------------------------------------------------------------------- /models/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/models/loss.py -------------------------------------------------------------------------------- /models/loss_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/models/loss_utils.py -------------------------------------------------------------------------------- /models/minkfpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/models/minkfpn.py -------------------------------------------------------------------------------- /models/minkgl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/models/minkgl.py -------------------------------------------------------------------------------- /models/minkloc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/models/minkloc.py -------------------------------------------------------------------------------- /models/minkloc3d_mulran.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/models/minkloc3d_mulran.txt -------------------------------------------------------------------------------- /models/model_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/models/model_factory.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/models/resnet.py -------------------------------------------------------------------------------- /third_party/minkloc3d/minkloc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/third_party/minkloc3d/minkloc.py -------------------------------------------------------------------------------- /third_party/pypcd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/third_party/pypcd.py -------------------------------------------------------------------------------- /third_party/scan_context/evaluate_scan_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/third_party/scan_context/evaluate_scan_context.py -------------------------------------------------------------------------------- /third_party/scan_context/scan_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/third_party/scan_context/scan_context.py -------------------------------------------------------------------------------- /training/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/training/train.py -------------------------------------------------------------------------------- /training/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/training/trainer.py -------------------------------------------------------------------------------- /weights/model_egonn_20210916_1104.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jac99/Egonn/HEAD/weights/model_egonn_20210916_1104.pth --------------------------------------------------------------------------------