├── .gitignore ├── LICENSE.md ├── README.md ├── datasets ├── demon_dataset.py ├── gta_sfm_dataset.py ├── multi_view_stereo_dataset.py └── stereo_dataset.py ├── multi_view_stereonet ├── __init__.py ├── losses.py ├── multi_view_stereonet.py └── multi_view_stereonet_utils.py ├── pretrained ├── demon_45epochs │ ├── checkpoints │ │ └── epoch0044 │ │ │ ├── checkpoint.tar │ │ │ └── stereo_network.pt │ ├── logs.txt │ ├── losses.txt │ ├── params.yaml │ └── validation.txt └── gta_sfm_150epochs │ ├── checkpoints │ └── epoch0149 │ │ ├── checkpoint.tar │ │ └── stereo_network.pt │ ├── logs.txt │ ├── losses.txt │ ├── params.yaml │ └── validation.txt ├── requirements.txt ├── scripts ├── demon │ ├── download_testdata.sh │ ├── download_traindata.sh │ ├── preparedata_test.py │ ├── preparedata_train.py │ └── requirements.txt └── gta_sfm │ ├── download.sh │ ├── extract.py │ └── requirements.txt ├── splits ├── demon_dpsnet_train.txt ├── demon_dpsnet_val.txt ├── demon_test.txt ├── demon_train.txt ├── demon_val.txt ├── gta_sfm_overlap0.5_1cmps_test.txt ├── gta_sfm_overlap0.5_2cmps_test.txt ├── gta_sfm_overlap0.5_3cmps_test.txt ├── gta_sfm_overlap0.5_4cmps_test.txt ├── gta_sfm_overlap0.5_5cmps_test.txt ├── gta_sfm_overlap0.5_test.txt └── gta_sfm_overlap0.5_train.txt ├── stereo ├── __init__.py └── image_predictor.py ├── test.py └── utils ├── __init__.py ├── depthmap_utils.py ├── image_gallery.py ├── image_utils.py ├── logger.py ├── losses.py ├── pfm_utils.py ├── pytorch_utils.py ├── resnet.py └── visualization.py /.gitignore: -------------------------------------------------------------------------------- 1 | *~ 2 | *pyc -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/README.md -------------------------------------------------------------------------------- /datasets/demon_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/datasets/demon_dataset.py -------------------------------------------------------------------------------- /datasets/gta_sfm_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/datasets/gta_sfm_dataset.py -------------------------------------------------------------------------------- /datasets/multi_view_stereo_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/datasets/multi_view_stereo_dataset.py -------------------------------------------------------------------------------- /datasets/stereo_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/datasets/stereo_dataset.py -------------------------------------------------------------------------------- /multi_view_stereonet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /multi_view_stereonet/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/multi_view_stereonet/losses.py -------------------------------------------------------------------------------- /multi_view_stereonet/multi_view_stereonet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/multi_view_stereonet/multi_view_stereonet.py -------------------------------------------------------------------------------- /multi_view_stereonet/multi_view_stereonet_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/multi_view_stereonet/multi_view_stereonet_utils.py -------------------------------------------------------------------------------- /pretrained/demon_45epochs/checkpoints/epoch0044/checkpoint.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/pretrained/demon_45epochs/checkpoints/epoch0044/checkpoint.tar -------------------------------------------------------------------------------- /pretrained/demon_45epochs/checkpoints/epoch0044/stereo_network.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/pretrained/demon_45epochs/checkpoints/epoch0044/stereo_network.pt -------------------------------------------------------------------------------- /pretrained/demon_45epochs/logs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/pretrained/demon_45epochs/logs.txt -------------------------------------------------------------------------------- /pretrained/demon_45epochs/losses.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/pretrained/demon_45epochs/losses.txt -------------------------------------------------------------------------------- /pretrained/demon_45epochs/params.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/pretrained/demon_45epochs/params.yaml -------------------------------------------------------------------------------- /pretrained/demon_45epochs/validation.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/pretrained/demon_45epochs/validation.txt -------------------------------------------------------------------------------- /pretrained/gta_sfm_150epochs/checkpoints/epoch0149/checkpoint.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/pretrained/gta_sfm_150epochs/checkpoints/epoch0149/checkpoint.tar -------------------------------------------------------------------------------- /pretrained/gta_sfm_150epochs/checkpoints/epoch0149/stereo_network.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/pretrained/gta_sfm_150epochs/checkpoints/epoch0149/stereo_network.pt -------------------------------------------------------------------------------- /pretrained/gta_sfm_150epochs/logs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/pretrained/gta_sfm_150epochs/logs.txt -------------------------------------------------------------------------------- /pretrained/gta_sfm_150epochs/losses.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/pretrained/gta_sfm_150epochs/losses.txt -------------------------------------------------------------------------------- /pretrained/gta_sfm_150epochs/params.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/pretrained/gta_sfm_150epochs/params.yaml -------------------------------------------------------------------------------- /pretrained/gta_sfm_150epochs/validation.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/pretrained/gta_sfm_150epochs/validation.txt -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/demon/download_testdata.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/scripts/demon/download_testdata.sh -------------------------------------------------------------------------------- /scripts/demon/download_traindata.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/scripts/demon/download_traindata.sh -------------------------------------------------------------------------------- /scripts/demon/preparedata_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/scripts/demon/preparedata_test.py -------------------------------------------------------------------------------- /scripts/demon/preparedata_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/scripts/demon/preparedata_train.py -------------------------------------------------------------------------------- /scripts/demon/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/scripts/demon/requirements.txt -------------------------------------------------------------------------------- /scripts/gta_sfm/download.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/scripts/gta_sfm/download.sh -------------------------------------------------------------------------------- /scripts/gta_sfm/extract.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/scripts/gta_sfm/extract.py -------------------------------------------------------------------------------- /scripts/gta_sfm/requirements.txt: -------------------------------------------------------------------------------- 1 | Pillow 2 | numpy 3 | h5py 4 | -------------------------------------------------------------------------------- /splits/demon_dpsnet_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/splits/demon_dpsnet_train.txt -------------------------------------------------------------------------------- /splits/demon_dpsnet_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/splits/demon_dpsnet_val.txt -------------------------------------------------------------------------------- /splits/demon_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/splits/demon_test.txt -------------------------------------------------------------------------------- /splits/demon_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/splits/demon_train.txt -------------------------------------------------------------------------------- /splits/demon_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/splits/demon_val.txt -------------------------------------------------------------------------------- /splits/gta_sfm_overlap0.5_1cmps_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/splits/gta_sfm_overlap0.5_1cmps_test.txt -------------------------------------------------------------------------------- /splits/gta_sfm_overlap0.5_2cmps_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/splits/gta_sfm_overlap0.5_2cmps_test.txt -------------------------------------------------------------------------------- /splits/gta_sfm_overlap0.5_3cmps_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/splits/gta_sfm_overlap0.5_3cmps_test.txt -------------------------------------------------------------------------------- /splits/gta_sfm_overlap0.5_4cmps_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/splits/gta_sfm_overlap0.5_4cmps_test.txt -------------------------------------------------------------------------------- /splits/gta_sfm_overlap0.5_5cmps_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/splits/gta_sfm_overlap0.5_5cmps_test.txt -------------------------------------------------------------------------------- /splits/gta_sfm_overlap0.5_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/splits/gta_sfm_overlap0.5_test.txt -------------------------------------------------------------------------------- /splits/gta_sfm_overlap0.5_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/splits/gta_sfm_overlap0.5_train.txt -------------------------------------------------------------------------------- /stereo/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /stereo/image_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/stereo/image_predictor.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/test.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/depthmap_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/utils/depthmap_utils.py -------------------------------------------------------------------------------- /utils/image_gallery.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/utils/image_gallery.py -------------------------------------------------------------------------------- /utils/image_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/utils/image_utils.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/utils/losses.py -------------------------------------------------------------------------------- /utils/pfm_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/utils/pfm_utils.py -------------------------------------------------------------------------------- /utils/pytorch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/utils/pytorch_utils.py -------------------------------------------------------------------------------- /utils/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/utils/resnet.py -------------------------------------------------------------------------------- /utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/robustrobotics/multi_view_stereonet/HEAD/utils/visualization.py --------------------------------------------------------------------------------