Python scripts form performing stereo depth estimation using the high res stereo model in PyTorch.

Stereo depth estimation on the cones images from the Middlebury dataset (https://vision.middlebury.edu/stereo/data/scenes2003/)

• OpenCV, imread-from-url and pytorch. Also, pafy and youtube-dl are required for youtube video inference.

pip install -r requirements.txt
pip install pafy youtube-dl

Pytorch: Check the Pytorch website to find the best method to install Pytorch in your computer.

Download the pretrained model from the original repository and save it into the models folder.

• Image inference:

python imageDepthEstimation.py 

• Video inference:

python videoDepthEstimation.py

• DrivingStereo dataset inference:

python drivingStereoTest.py

• High Res Stereo model: https://github.com/gengshan-y/high-res-stereo
• DrivingStereo dataset: https://drivingstereo-dataset.github.io/
• Original paper: https://arxiv.org/abs/1912.06704