Home

Awesome

TC-Depth

Project Page | arXiv | Workshop

Official Repository for

Attention meets Geometry: Geometry Guided Spatial-Temporal Attention for Consistent Self-Supervised Monocular Depth Estimation

Daoyi Gao, Hanzhi Chen, Patrick Ruhkamp, Nassir Navab, Benjamin Busam - 3DV, 2021.

Spatial-Temporal Attention through Self-Supervised Geometric Guidance

Daoyi Gao, Hanzhi Chen, Patrick Ruhkamp, Nassir Navab, Benjamin Busam - ICCV Workshop on Self-supervised Learning for Next-Generation Industry-level Autonomous Driving, 2021.

Bold: equal contribution

<div align=center><img src="resources/09260052_new.gif"/></div>

🤓 TL;DR

🎇 News

🖊 Results

Qualitative Results

<p align="center"> <img src="resources/teaser.png" alt="teaser figure" width="200" /> <img src="resources/reconstruction.png" alt="reconstruction figure" width="600" /> </p>

Spatial-Temporal Attention

<p align="center"> <img src="resources/attention.png" alt="teaser figure" width="800" /> </p>

Temporal Consistency Metric (TCM)

<p align="center"> <img src="resources/tcm.png" alt="tcm visualisation" width="400" /> </p>

💽 Resource

GT for TCM

3 Frames Track | 5 Frames Track | 7 Frames Track

Pretrained Weights

KITTI DRNC-26

📄 Citation

If you find our work useful, please consider citing the following papers:

@inproceedings{ruhkamp2021attention,
    title = {Attention meets Geometry: Geometry Guided Spatial-Temporal Attention for Consistent Self-Supervised Monocular Depth Estimation},
    author = {Patrick Ruhkamp and
              Daoyi Geo and
              Hanzhi Chen and
              Nassir Navab and
              Benjamin Busam},
    booktitle = {IEEE International Conference on 3D Vision (3DV)},
    year = {2021},
    month = {December}
}

@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}
}

❤ Acknowledgement

Our implementation is based on MonoDepth2 and follows their code structure. Thanks for their great contribution :)