HITNET-Stereo-Depth-estimation

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

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

Requirements

• OpenCV, numpy and tensorflo. pafy 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

Inference video Example

References:

• Hitnet model: https://github.com/google-research/google-research/tree/master/hitnet
• DrivingStereo dataset: https://drivingstereo-dataset.github.io/
• Original paper: https://arxiv.org/abs/2007.12140