Tensorflow lite - depth map - based on stereo vision

1k Views Asked by At

I have not found any project, library, model or guide to measure distance using stereo imaging with tensorflow lite 2.

I want to be able to measure distance from stereo images. I would like to be able to run it on android, so I would like to create a model in tensorflow 2 or use an existing one. But I don't know where to start, everything I found uses pytorch

I know that opencv provides a method to do it, but according to the literature neural networks achieve better results. But I have not found any model in tensorflow 2. I can't use google ar, because my device doesn't support google play services.

1

There are 1 best solutions below

1
On

I have just uploaded a repository for this purpose: https://github.com/ibaiGorordo/TFLite-HITNET-Stereo-depth-estimation

It uses the stereo depth estimation HITNET model (from Google-Research) converted to tensorflow lite by @PINTO0309 (find the models here: https://github.com/PINTO0309/PINTO_model_zoo/tree/main/142_HITNET)