├── .gitignore ├── LICENSE ├── README.md ├── ace ├── ace_network.py ├── ace_trainer.py ├── ace_util.py ├── ace_vis_util.py └── ace_visualizer.py ├── ace_encoder_pretrained.pt ├── ace_head_dummy.pt ├── dataset.py ├── datasets ├── dataset_util.py ├── setup_12scenes.py ├── setup_7scenes.py └── setup_wayspots.py ├── dsacstar ├── dsacstar.cpp ├── dsacstar_derivative.h ├── dsacstar_loss.h ├── dsacstar_types.h ├── dsacstar_util.h ├── dsacstar_util_rgbd.h ├── setup.py ├── stop_watch.h ├── thread_rand.cpp └── thread_rand.h ├── environment.yml ├── imgs └── marepo.png ├── loss.py ├── marepo ├── __init__.py ├── callbacks.py ├── dataset_marepo.py ├── load_dataset_marepo.py ├── marepo_network.py ├── marepo_trainer.py ├── marepo_vis_util.py ├── metrics.py └── util.py ├── opt_marepo.py ├── plot_preprocess.py ├── preprocess_marepo.py ├── preprocess_scripts ├── create_7scenes_finetune_testset.sh ├── create_mapfree_all_dataset.sh ├── create_wayspots_testset.sh └── flip_dataset.sh ├── read_log_marepo.py ├── scripts ├── test_12scenes.sh ├── test_7scenes.sh ├── test_wayspots.sh ├── train_mapfree.sh ├── train_marepo.sh ├── train_marepo_s_7scenes.sh └── train_marepo_s_wayspots.sh ├── test_ace.py ├── test_marepo.py ├── test_marepo_util.py ├── train_ace.py ├── train_marepo.py └── transformer ├── __init__.py ├── config ├── nerf_focal_12T1R_256_homo.json ├── nerf_focal_12T1R_256_homo_9D.json ├── nerf_focal_12T1R_256_homo_c2f.json └── nerf_focal_12T1R_256_homo_c2f_9D.json ├── linear_attention.py ├── position_encoding.py ├── position_encoding_nerf.py └── transformer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/README.md -------------------------------------------------------------------------------- /ace/ace_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/ace/ace_network.py -------------------------------------------------------------------------------- /ace/ace_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/ace/ace_trainer.py -------------------------------------------------------------------------------- /ace/ace_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/ace/ace_util.py -------------------------------------------------------------------------------- /ace/ace_vis_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/ace/ace_vis_util.py -------------------------------------------------------------------------------- /ace/ace_visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/ace/ace_visualizer.py -------------------------------------------------------------------------------- /ace_encoder_pretrained.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/ace_encoder_pretrained.pt -------------------------------------------------------------------------------- /ace_head_dummy.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/ace_head_dummy.pt -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/dataset.py -------------------------------------------------------------------------------- /datasets/dataset_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/datasets/dataset_util.py -------------------------------------------------------------------------------- /datasets/setup_12scenes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/datasets/setup_12scenes.py -------------------------------------------------------------------------------- /datasets/setup_7scenes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/datasets/setup_7scenes.py -------------------------------------------------------------------------------- /datasets/setup_wayspots.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/datasets/setup_wayspots.py -------------------------------------------------------------------------------- /dsacstar/dsacstar.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/dsacstar/dsacstar.cpp -------------------------------------------------------------------------------- /dsacstar/dsacstar_derivative.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/dsacstar/dsacstar_derivative.h -------------------------------------------------------------------------------- /dsacstar/dsacstar_loss.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/dsacstar/dsacstar_loss.h -------------------------------------------------------------------------------- /dsacstar/dsacstar_types.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/dsacstar/dsacstar_types.h -------------------------------------------------------------------------------- /dsacstar/dsacstar_util.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/dsacstar/dsacstar_util.h -------------------------------------------------------------------------------- /dsacstar/dsacstar_util_rgbd.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/dsacstar/dsacstar_util_rgbd.h -------------------------------------------------------------------------------- /dsacstar/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/dsacstar/setup.py -------------------------------------------------------------------------------- /dsacstar/stop_watch.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/dsacstar/stop_watch.h -------------------------------------------------------------------------------- /dsacstar/thread_rand.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/dsacstar/thread_rand.cpp -------------------------------------------------------------------------------- /dsacstar/thread_rand.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/dsacstar/thread_rand.h -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/environment.yml -------------------------------------------------------------------------------- /imgs/marepo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/imgs/marepo.png -------------------------------------------------------------------------------- /loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/loss.py -------------------------------------------------------------------------------- /marepo/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /marepo/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/marepo/callbacks.py -------------------------------------------------------------------------------- /marepo/dataset_marepo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/marepo/dataset_marepo.py -------------------------------------------------------------------------------- /marepo/load_dataset_marepo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/marepo/load_dataset_marepo.py -------------------------------------------------------------------------------- /marepo/marepo_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/marepo/marepo_network.py -------------------------------------------------------------------------------- /marepo/marepo_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/marepo/marepo_trainer.py -------------------------------------------------------------------------------- /marepo/marepo_vis_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/marepo/marepo_vis_util.py -------------------------------------------------------------------------------- /marepo/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/marepo/metrics.py -------------------------------------------------------------------------------- /marepo/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/marepo/util.py -------------------------------------------------------------------------------- /opt_marepo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/opt_marepo.py -------------------------------------------------------------------------------- /plot_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/plot_preprocess.py -------------------------------------------------------------------------------- /preprocess_marepo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/preprocess_marepo.py -------------------------------------------------------------------------------- /preprocess_scripts/create_7scenes_finetune_testset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/preprocess_scripts/create_7scenes_finetune_testset.sh -------------------------------------------------------------------------------- /preprocess_scripts/create_mapfree_all_dataset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/preprocess_scripts/create_mapfree_all_dataset.sh -------------------------------------------------------------------------------- /preprocess_scripts/create_wayspots_testset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/preprocess_scripts/create_wayspots_testset.sh -------------------------------------------------------------------------------- /preprocess_scripts/flip_dataset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/preprocess_scripts/flip_dataset.sh -------------------------------------------------------------------------------- /read_log_marepo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/read_log_marepo.py -------------------------------------------------------------------------------- /scripts/test_12scenes.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/scripts/test_12scenes.sh -------------------------------------------------------------------------------- /scripts/test_7scenes.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/scripts/test_7scenes.sh -------------------------------------------------------------------------------- /scripts/test_wayspots.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/scripts/test_wayspots.sh -------------------------------------------------------------------------------- /scripts/train_mapfree.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/scripts/train_mapfree.sh -------------------------------------------------------------------------------- /scripts/train_marepo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/scripts/train_marepo.sh -------------------------------------------------------------------------------- /scripts/train_marepo_s_7scenes.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/scripts/train_marepo_s_7scenes.sh -------------------------------------------------------------------------------- /scripts/train_marepo_s_wayspots.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/scripts/train_marepo_s_wayspots.sh -------------------------------------------------------------------------------- /test_ace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/test_ace.py -------------------------------------------------------------------------------- /test_marepo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/test_marepo.py -------------------------------------------------------------------------------- /test_marepo_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/test_marepo_util.py -------------------------------------------------------------------------------- /train_ace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/train_ace.py -------------------------------------------------------------------------------- /train_marepo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/train_marepo.py -------------------------------------------------------------------------------- /transformer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /transformer/config/nerf_focal_12T1R_256_homo.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/transformer/config/nerf_focal_12T1R_256_homo.json -------------------------------------------------------------------------------- /transformer/config/nerf_focal_12T1R_256_homo_9D.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/transformer/config/nerf_focal_12T1R_256_homo_9D.json -------------------------------------------------------------------------------- /transformer/config/nerf_focal_12T1R_256_homo_c2f.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/transformer/config/nerf_focal_12T1R_256_homo_c2f.json -------------------------------------------------------------------------------- /transformer/config/nerf_focal_12T1R_256_homo_c2f_9D.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/transformer/config/nerf_focal_12T1R_256_homo_c2f_9D.json -------------------------------------------------------------------------------- /transformer/linear_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/transformer/linear_attention.py -------------------------------------------------------------------------------- /transformer/position_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/transformer/position_encoding.py -------------------------------------------------------------------------------- /transformer/position_encoding_nerf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/transformer/position_encoding_nerf.py -------------------------------------------------------------------------------- /transformer/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nianticlabs/marepo/HEAD/transformer/transformer.py --------------------------------------------------------------------------------