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