This is the repository for our submission "SPDET: Edge-Aware Self-Supervised Panoramic Depth Estimation Transformer With Spherical Geometry".

Requirements

pytorch==1.11.0
cudatoolkit=10.2
numpy==1.21.5
opencv-python==4.6.0.66
timm==0.6.7

Usage

1. Train and test the model with the script:

    ./spdet.sh
   

2. The pretrained model parameters are accessed in this link.

TODO

• more detailed training instructions
• test script for custom input panoramas