This repository builds upon the open source pytorch Active stereo net implementation and extends it to active tartanair and D435i datsets.
ActiveStereoNet: End-to-End Self-Supervised Learning for Active Stereo Systems
Datasets used:
Please, use the links provided to download the datasets and update the data_root field in the Options/*.json files.
To train on D435i sequences:
sh d435i.sh
To train on Active Tartanair sequences:
sh tartanair.sh