Skip to content

Python scripts for performing stereo depth estimation using the HITNET Tensorflow model.

License

Notifications You must be signed in to change notification settings

ux3d/HITNET-Stereo-Depth-estimation

 
 

Repository files navigation

HITNET-Stereo-Depth-estimation

Python scripts for performing stereo depth estimation using the HITNET Tensorflow model from Google Research.

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

Requirements

  • OpenCV, numpy and tensorflo. pafy (pip install git+https://github.com/zizo-pro/pafy@b8976f22c19e4ab5515cacbfae0a3970370c102b) and youtube-dl are required for youtube video inference.
  • For the drivingStereo dataset, download the data from: https://drivingstereo-dataset.github.io/

Tensorflow models

Download the tensorflow models from the original repository and save them into the models folder.

Examples

  • Image inference:
python imageDepthEstimation.py 
  • Video inference:
python videoDepthEstimation.py
  • DrivingStereo dataset inference:
python drivingStereoTest.py

!Hitnet stereo depth estimation on video

References:

About

Python scripts for performing stereo depth estimation using the HITNET Tensorflow model.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%