├── LICENSE.md ├── README.md ├── code ├── DSO │ ├── __init__.py │ ├── cam_info_7scenes.mat │ ├── cam_info_kitti.mat │ ├── cam_info_scanNet.mat │ └── dso_io.py ├── ICP │ ├── __init__.py │ └── opt_pose_numerical.py ├── local_test.sh ├── local_test_LBA.sh ├── local_train_kitti.sh ├── local_train_scanNet.sh ├── mdataloader │ ├── __init__.py │ ├── batch_loader.py │ ├── dl_7scenes.py │ ├── kitti.py │ ├── kitti_split │ │ ├── test_eigen.txt │ │ ├── testing.txt │ │ ├── train_eigen.txt │ │ └── training.txt │ ├── m_preprocess.py │ ├── mdata.py │ ├── misc.py │ ├── scanNet.py │ └── scanNet_split │ │ ├── scannet_rob_test.txt │ │ ├── scannet_test.txt │ │ ├── scannet_train.txt │ │ ├── scannet_val.txt │ │ └── single.txt ├── mio │ ├── __init__.py │ ├── imgIO.py │ └── vidIO.py ├── models │ ├── GF │ │ ├── __init__.py │ │ ├── box_filter.py │ │ └── guided_filter.py │ ├── KVNET.py │ ├── Refine.py │ ├── __init__.py │ ├── basic.py │ ├── m_submodule.py │ └── psm_submodule.py ├── mutils │ ├── __init__.py │ └── misc.py ├── run_demo.sh ├── run_dso.py ├── saved_models │ └── download_weights.sh ├── test_KVNet.py ├── test_KVNet_LBA.py ├── test_utils │ ├── __init__.py │ ├── export_res.py │ └── test_KVNet.py ├── train_KVNet.py ├── train_utils │ ├── Logger.py │ ├── __init__.py │ └── train_KVNet.py ├── utils │ ├── __init__.py │ ├── models.py │ ├── preprocess.py │ └── readpfm.py └── warping │ ├── View.py │ ├── __init__.py │ └── homography.py ├── data ├── download_LBA_demo_data.sh └── download_demo_data.sh ├── docs ├── LBA.md ├── TE.md └── TR.md ├── requirements.txt └── third_party ├── README.md ├── SensReader ├── Makefile ├── README.md ├── README.txt ├── decode.py ├── scannet_test.txt ├── scannet_train.txt ├── scannet_val.txt ├── sens.sln ├── sens.vcxproj ├── sens.vcxproj.filters └── src │ ├── main.cpp │ ├── sensorData.h │ └── sensorData │ ├── stb_image.h │ └── stb_image_write.h ├── nnDepthDSO.patch └── setup_dso.sh /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/README.md -------------------------------------------------------------------------------- /code/DSO/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | -------------------------------------------------------------------------------- /code/DSO/cam_info_7scenes.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/DSO/cam_info_7scenes.mat -------------------------------------------------------------------------------- /code/DSO/cam_info_kitti.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/DSO/cam_info_kitti.mat -------------------------------------------------------------------------------- /code/DSO/cam_info_scanNet.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/DSO/cam_info_scanNet.mat -------------------------------------------------------------------------------- /code/DSO/dso_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/DSO/dso_io.py -------------------------------------------------------------------------------- /code/ICP/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | -------------------------------------------------------------------------------- /code/ICP/opt_pose_numerical.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/ICP/opt_pose_numerical.py -------------------------------------------------------------------------------- /code/local_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/local_test.sh -------------------------------------------------------------------------------- /code/local_test_LBA.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/local_test_LBA.sh -------------------------------------------------------------------------------- /code/local_train_kitti.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/local_train_kitti.sh -------------------------------------------------------------------------------- /code/local_train_scanNet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/local_train_scanNet.sh -------------------------------------------------------------------------------- /code/mdataloader/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | 3 | -------------------------------------------------------------------------------- /code/mdataloader/batch_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/mdataloader/batch_loader.py -------------------------------------------------------------------------------- /code/mdataloader/dl_7scenes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/mdataloader/dl_7scenes.py -------------------------------------------------------------------------------- /code/mdataloader/kitti.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/mdataloader/kitti.py -------------------------------------------------------------------------------- /code/mdataloader/kitti_split/test_eigen.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/mdataloader/kitti_split/test_eigen.txt -------------------------------------------------------------------------------- /code/mdataloader/kitti_split/testing.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/mdataloader/kitti_split/testing.txt -------------------------------------------------------------------------------- /code/mdataloader/kitti_split/train_eigen.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/mdataloader/kitti_split/train_eigen.txt -------------------------------------------------------------------------------- /code/mdataloader/kitti_split/training.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/mdataloader/kitti_split/training.txt -------------------------------------------------------------------------------- /code/mdataloader/m_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/mdataloader/m_preprocess.py -------------------------------------------------------------------------------- /code/mdataloader/mdata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/mdataloader/mdata.py -------------------------------------------------------------------------------- /code/mdataloader/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/mdataloader/misc.py -------------------------------------------------------------------------------- /code/mdataloader/scanNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/mdataloader/scanNet.py -------------------------------------------------------------------------------- /code/mdataloader/scanNet_split/scannet_rob_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/mdataloader/scanNet_split/scannet_rob_test.txt -------------------------------------------------------------------------------- /code/mdataloader/scanNet_split/scannet_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/mdataloader/scanNet_split/scannet_test.txt -------------------------------------------------------------------------------- /code/mdataloader/scanNet_split/scannet_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/mdataloader/scanNet_split/scannet_train.txt -------------------------------------------------------------------------------- /code/mdataloader/scanNet_split/scannet_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/mdataloader/scanNet_split/scannet_val.txt -------------------------------------------------------------------------------- /code/mdataloader/scanNet_split/single.txt: -------------------------------------------------------------------------------- 1 | scene0534_00 2 | -------------------------------------------------------------------------------- /code/mio/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/mio/imgIO.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/mio/imgIO.py -------------------------------------------------------------------------------- /code/mio/vidIO.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/mio/vidIO.py -------------------------------------------------------------------------------- /code/models/GF/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | -------------------------------------------------------------------------------- /code/models/GF/box_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/models/GF/box_filter.py -------------------------------------------------------------------------------- /code/models/GF/guided_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/models/GF/guided_filter.py -------------------------------------------------------------------------------- /code/models/KVNET.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/models/KVNET.py -------------------------------------------------------------------------------- /code/models/Refine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/models/Refine.py -------------------------------------------------------------------------------- /code/models/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | -------------------------------------------------------------------------------- /code/models/basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/models/basic.py -------------------------------------------------------------------------------- /code/models/m_submodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/models/m_submodule.py -------------------------------------------------------------------------------- /code/models/psm_submodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/models/psm_submodule.py -------------------------------------------------------------------------------- /code/mutils/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | -------------------------------------------------------------------------------- /code/mutils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/mutils/misc.py -------------------------------------------------------------------------------- /code/run_demo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/run_demo.sh -------------------------------------------------------------------------------- /code/run_dso.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/run_dso.py -------------------------------------------------------------------------------- /code/saved_models/download_weights.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/saved_models/download_weights.sh -------------------------------------------------------------------------------- /code/test_KVNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/test_KVNet.py -------------------------------------------------------------------------------- /code/test_KVNet_LBA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/test_KVNet_LBA.py -------------------------------------------------------------------------------- /code/test_utils/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | 6 | -------------------------------------------------------------------------------- /code/test_utils/export_res.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/test_utils/export_res.py -------------------------------------------------------------------------------- /code/test_utils/test_KVNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/test_utils/test_KVNet.py -------------------------------------------------------------------------------- /code/train_KVNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/train_KVNet.py -------------------------------------------------------------------------------- /code/train_utils/Logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/train_utils/Logger.py -------------------------------------------------------------------------------- /code/train_utils/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | 3 | 4 | 5 | -------------------------------------------------------------------------------- /code/train_utils/train_KVNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/train_utils/train_KVNet.py -------------------------------------------------------------------------------- /code/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/utils/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/utils/models.py -------------------------------------------------------------------------------- /code/utils/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/utils/preprocess.py -------------------------------------------------------------------------------- /code/utils/readpfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/utils/readpfm.py -------------------------------------------------------------------------------- /code/warping/View.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/warping/View.py -------------------------------------------------------------------------------- /code/warping/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | 3 | -------------------------------------------------------------------------------- /code/warping/homography.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/code/warping/homography.py -------------------------------------------------------------------------------- /data/download_LBA_demo_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/data/download_LBA_demo_data.sh -------------------------------------------------------------------------------- /data/download_demo_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/data/download_demo_data.sh -------------------------------------------------------------------------------- /docs/LBA.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/docs/LBA.md -------------------------------------------------------------------------------- /docs/TE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/docs/TE.md -------------------------------------------------------------------------------- /docs/TR.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/docs/TR.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/requirements.txt -------------------------------------------------------------------------------- /third_party/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/third_party/README.md -------------------------------------------------------------------------------- /third_party/SensReader/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/third_party/SensReader/Makefile -------------------------------------------------------------------------------- /third_party/SensReader/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/third_party/SensReader/README.md -------------------------------------------------------------------------------- /third_party/SensReader/README.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/third_party/SensReader/README.txt -------------------------------------------------------------------------------- /third_party/SensReader/decode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/third_party/SensReader/decode.py -------------------------------------------------------------------------------- /third_party/SensReader/scannet_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/third_party/SensReader/scannet_test.txt -------------------------------------------------------------------------------- /third_party/SensReader/scannet_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/third_party/SensReader/scannet_train.txt -------------------------------------------------------------------------------- /third_party/SensReader/scannet_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/third_party/SensReader/scannet_val.txt -------------------------------------------------------------------------------- /third_party/SensReader/sens.sln: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/third_party/SensReader/sens.sln -------------------------------------------------------------------------------- /third_party/SensReader/sens.vcxproj: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/third_party/SensReader/sens.vcxproj -------------------------------------------------------------------------------- /third_party/SensReader/sens.vcxproj.filters: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/third_party/SensReader/sens.vcxproj.filters -------------------------------------------------------------------------------- /third_party/SensReader/src/main.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/third_party/SensReader/src/main.cpp -------------------------------------------------------------------------------- /third_party/SensReader/src/sensorData.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/third_party/SensReader/src/sensorData.h -------------------------------------------------------------------------------- /third_party/SensReader/src/sensorData/stb_image.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/third_party/SensReader/src/sensorData/stb_image.h -------------------------------------------------------------------------------- /third_party/SensReader/src/sensorData/stb_image_write.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/third_party/SensReader/src/sensorData/stb_image_write.h -------------------------------------------------------------------------------- /third_party/nnDepthDSO.patch: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/third_party/nnDepthDSO.patch -------------------------------------------------------------------------------- /third_party/setup_dso.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/neuralrgbd/HEAD/third_party/setup_dso.sh --------------------------------------------------------------------------------