Python scripts for performing stereo depth estimation using the HITNET Tensorflow model.
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/)
pip install git+https://github.com/zizo-pro/pafy@b8976f22c19e4ab5515cacbfae0a3970370c102b
) and youtube-dl are required for youtube video inference.Download the tensorflow models from the original repository and save them into the models folder.
python imageDepthEstimation.py
python videoDepthEstimation.py
python drivingStereoTest.py