/PyTorch-High-Res-Stereo-Depth-Estimation

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

Primary LanguagePythonMIT LicenseMIT

PyTorch-High-Res-Stereo-Depth-Estimation

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

High res stereo depth estimation 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

High res stereo depth estimation Pytorch

References: