PyTorch-High-Res-Stereo-Depth-Estimation

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/)

Requirements

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

Installation

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.

Pretrained model

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

Examples

• Image inference:

python imageDepthEstimation.py 

• Video inference:

python videoDepthEstimation.py

• DrivingStereo dataset inference:

python drivingStereoTest.py

Inference video Example

References:

• 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