Python scripts for performing stereo depth estimation using the MobileStereoNet model in ONNX
Python scripts for performing stereo depth estimation using the MobileStereoNet model in ONNX
Stereo depth estimation on the cones images from the Middlebury dataset (https://vision.middlebury.edu/stereo/data/scenes2003/)
- Check the requirements.txt file. Additionally, pafy and youtube-dl are required for youtube video inference.
- DrivingStereo dataset, ONLY for the
driving_sereo_test.pyscript. Link: https://drivingstereo-dataset.github.io/
pip install -r requirements.txt
pip install pafy youtube-dl
The original models were converted to different formats (including .onnx) by PINTO0309, the models can be found in his repository.
The Pytorch pretrained model was taken from the original repository.