High Quality Monocular Depth Estimation via Transfer Learning
GPL-3.0 License
Ibraheem Alhashim and Peter Wonka
[Update] Our latest method with better performance can be found here AdaBins.
Offical Keras (TensorFlow) implementaiton. If you have any questions or need more help with the code, contact the first author.
[Update] Added a Colab notebook to try the method on the fly.
[Update] Experimental TensorFlow 2.0 implementation added.
[Update] Experimental PyTorch code added.
keras pillow matplotlib scikit-learn scikit-image opencv-python pydot
and GraphViz
for the model graph visualization and PyGLM PySide2 pyopengl
for the GUI demo.python test.py
. You should see a montage of images with their estimated depth maps.python demo.py
. It requires the packages PyGLM PySide2 pyopengl
.python train.py --data nyu --gpus 4 --bs 8
.python evaluate.py
.Corresponding paper to cite:
@article{Alhashim2018,
author = {Ibraheem Alhashim and Peter Wonka},
title = {High Quality Monocular Depth Estimation via Transfer Learning},
journal = {arXiv e-prints},
volume = {abs/1812.11941},
year = {2018},
url = {https://arxiv.org/abs/1812.11941},
eid = {arXiv:1812.11941},
eprint = {1812.11941}
}