Python scripts form performing stereo depth estimation using the HITNET model in ONNX.

ibaiGorordo ibaiGorordo Last update: Sep 12, 2022

ONNX-HITNET-Stereo-Depth-estimation

Python scripts form performing stereo depth estimation using the HITNET model in ONNX.

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

Requirements

Installation

pip install -r requirements.txtpip install pafy youtube-dl

ONNX model

The original models were converted to different formats (including .onnx) by PINTO0309, download the models from his repository and save them into the models folder.

Original Tensorflow model

The Tensorflow pretrained model was taken from the original repository.

Examples

  • Depthai OAK-D series inference on the host:
python depthai_host_depth_estimation.py
  • Image inference:
python image_depth_estimation.py 
  • Video inference:
python video_depth_estimation.py
  • DrivingStereo dataset inference:
python driving_stereo_test.py

Pytorch inference

For performing the inference in Tensorflow, check my other repository HITNET Stereo Depth estimation.

TFLite inference

For performing the inference in TFLite, check my other repository TFLite HITNET Stereo Depth estimation.

Inference video Example

Hitnet stereo depth estimation ONNX

References:

Screenshot of pybot with hacker theme

A Anti-DDoS script to protect Nginx web servers using Lua with a Javascript based authentication puzzle inspired by Cloudflare I am under attack mode I built my own Anti-DDoS authentication HTML page…

Subscribe to our newsletter