/monodepth2-TensorFlow2

TensorFlow 2 (Keras) implementation of Monodepth2, Monocular Depth Prediction.

Primary LanguagePythonOtherNOASSERTION

Monodepth2 - TensorFlow 2

This is a TensorFlow 2 implementation of the original Monodepth2 PyTorch model, using the Keras Functional API. An illustration of the model's overview is available at the model.png image.

Being such an implementation, it inherits all the copyrights and licensing accredited to the original authors at: https://github.com/nianticlabs/monodepth2

Most prominent is the restriction for non-commercial use only (refer to the LICENSE file).

This repo is intended for those who'd prefer the TensorFlow-Keras framework, as well as being able to export the model with TensorFlow lite for edge devices (see the quantized model attached).

The model is based on the pre-trained PyTorch encoder & decoder for the single image mono_640x192 model.

Inference example

Input/Outputs

Paper citation

Digging into Self-Supervised Monocular Depth Prediction

Clément Godard, Oisin Mac Aodha, Michael Firman and Gabriel J. Brostow

ICCV 2019

If you find their work useful in your research please consider citing their paper:

@article{monodepth2,
  title     = {Digging into Self-Supervised Monocular Depth Prediction},
  author    = {Cl{\'{e}}ment Godard and
               Oisin {Mac Aodha} and
               Michael Firman and
               Gabriel J. Brostow},
  booktitle = {The International Conference on Computer Vision (ICCV)},
  month = {October},
year = {2019}
}