Skip to content

Accurate and comprehensive 3D scene understanding and reasoning are pivotal for the development of robotic and autonomous driving systems. The study focuses on advancing 3D scene understanding in the scope of 3D Semantic Occupancy Prediction, emphasizing spatial and temporal reasoning as key components.

Notifications You must be signed in to change notification settings

cepdnaclk/e17-4yp-S2TPVFormer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

62 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation


S2TPVFormer: Improving 3D Semantic Occupancy Prediction using Spatiotemporal Transformers


Sathira Silva*, Savindu Wannigama*, Prof. Roshan Ragel $\dagger$, Gihan Jayatilaka $\ddagger$

* Equal contribution $\dagger$ Project Supervisor $\ddagger$ Project Co-supervisor

News

Introduction

Temporal reasoning holds equal importance to spatial reasoning in a cognitive perception system. In human perception, temporal information is crucial for identifying occluded objects and determining the motion state of entities. A system proficient in spatiotemporal reasoning excels in making inferences with high temporal coherence. While previous works emphasize the significance of temporal fusion in 3D object detection, earlier attempts at 3D Semantic Occupancy Prediction (3D SOP) often overlooked the value of incorporating temporal information. The current state-of-the-art in 3D SOP literature seldom exploits temporal cues. This is evident in TPVFormer’s SOP visualizations, where adjacent prediction frames lack temporal coherence as they rely solely on the current time step for semantic predictions.

This work introduces S2TPVFormer, a variant of TPVFormer, which utilizes a spatiotemporal transformer architecture inspired by BEVFormer, for dense and temporally coherent 3D semantic occupancy prediction. Leveraging TPV (Tri-Perspective View) representation, the model’s spatiotemporal encoder generates temporally rich embeddings, fostering coherent predictions. The study proposes a novel Temporal Cross-View Hybrid Attention mechanism, enabling the exchange of spatiotemporal information across different views. To illustrate the efficacy of temporal information incorporation and the potential of the new attention mechanism, the research explores three distinct temporal fusion paradigms.

Overview of our Contributions

To summarize, this work contributes in the following ways,

  • We pioneer the use of TPV representation for embedding spatiotemporal information in 3D scenes within the domain of vision-centric SOP and the broader 3D perception literature.
  • We introduce a novel temporal fusion workflow for TPV representation, analyzing how CVHA facilitates the sharing of spatiotemporal information across the three planes.
  • The lower parameter model of our method achieves a significant 3.1% improvement in mIoU for 3D SOP when evaluated on the nuScenes validation dataset with TPVFormer’s sparse pseudo-voxel ground truth, compared to TPVFormer.

Results

Related Projects

Our code is based on TPVFormer and BEVFormer. Many thanks to them!

Team

External Links

Citation

If this work is helpful for your research, please consider citing:

@article{silva2024s2tpvformer,
    title={S2TPVFormer: Spatio-Temporal Tri-Perspective View for temporally coherent 3D Semantic Occupancy Prediction},
    author={Sathira Silva and Savindu Bhashitha Wannigama and Roshan Ragel and Gihan Jayatilaka},
    journal={arXiv:2401.13785},
    year={2024},
}

About

Accurate and comprehensive 3D scene understanding and reasoning are pivotal for the development of robotic and autonomous driving systems. The study focuses on advancing 3D scene understanding in the scope of 3D Semantic Occupancy Prediction, emphasizing spatial and temporal reasoning as key components.

Resources

Stars

Watchers

Forks