SGFormer: Spherical Geometry Transformer for 360 Depth Estimation

Junsong Zhang, Zisong Chen, Chunyu Lin, Lang Nie, Zhijie Shen +3 more
4/23/2024
cs.CVcs.AI

No official code found. Best implementation: iuiuJaon/SGFormer (4 stars, last commit 2024-09-20).

Abstract

Panoramic distortion poses a significant challenge in 360 depth estimation, particularly pronounced at the north and south poles. Existing methods either adopt a bi-projection fusion strategy to remove distortions or model long-range dependencies to capture global structures, which can result in either unclear structure or insufficient local perception. In this paper, we propose a spherical geometry transformer, named SGFormer, to address the above issues, with an innovative step to integrate spherical geometric priors into vision transformers. To this end, we retarget the transformer decoder to a spherical prior decoder (termed SPDecoder), which endeavors to uphold the integrity of spherical structures during decoding. Concretely, we leverage bipolar re-projection, circular rotation, and curve local embedding to preserve the spherical characteristics of equidistortion, continuity, and surface distance, respectively. Furthermore, we present a query-based global conditional position embedding to compensate for spatial structure at varying resolutions. It not only boosts the global perception of spatial position but also sharpens the depth structure across different patches. Finally, we conduct extensive experiments on popular benchmarks, demonstrating our superiority over state-of-the-art solutions.

View on arXivView PDF

Code Implementations(1)

iuiuJaon/SGFormerCommunity58%

This is the PyTorch implementation of our new work SGFormer——Spherical Geometry Transformer for 360° Depth Estimation. The code will be made publicly available after acceptance.

40Sep 20, 20242 years ago

Similar Papers

Cite this paper

@article{zhang2024sgformer,
  title  = {SGFormer: Spherical Geometry Transformer for 360 Depth Estimation},
  author = {Junsong Zhang and Zisong Chen and Chunyu Lin and Lang Nie and Zhijie Shen and Kang Liao and Junda Huang and Yao Zhao},
  year   = {2024},
  eprint = {2404.14979},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2404.14979v3}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2404.14979.svg)](https://codeofpaper.com/paper/2404.14979)

Discussion