Skip to content

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

License

Notifications You must be signed in to change notification settings

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

Repository files navigation

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:

About

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

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages