4D-StOP: Panoptic Segmentation of 4D LiDAR using Spatio-temporal Object Proposal Generation and Aggregation

arXiv (Cornell University)

Abstract

In this work, we present a new paradigm, called 4D-StOP, to tackle the task of 4D Panoptic LiDAR Segmentation. 4D-StOP first generates spatio-temporal proposals using voting-based center predictions, where each point in the 4D volume votes for a corresponding center. These tracklet proposals are further aggregated using learned geometric features. The tracklet aggregation method effectively generates a video-level 4D scene representation over the entire space-time volume. This is in contrast to existing end-to-end trainable state-of-the-art approaches which use spatio-temporal embeddings that are represented by Gaussian probability distributions. Our voting-based tracklet generation method followed by geometric feature-based aggregation generates significantly improved panoptic LiDAR segmentation quality when compared to modeling the entire 4D volume using Gaussian probability distributions. 4D-StOP achieves a new state-of-the-art when applied to the SemanticKITTI test dataset with a score of 63.9 LSTQ, which is a large (+7%) improvement compared to current best-performing end-to-end trainable methods. The code and pre-trained models are available at: https://github.com/LarsKreuzberg/4D-StOP.

Authors 5

1. Lars Kreuzberg Aachen

   RWTH Aachen University

   Affiliation as printed

   RWTH Aachen University , Germany

2. Idil Esen Zulfikar Aachen

   RWTH Aachen University

   Affiliation as printed

   RWTH Aachen University , Germany

3. Sabarinath Mahadevan Aachen

   RWTH Aachen University

   Affiliation as printed

   RWTH Aachen University , Germany

4. Francis Engelmann

   ETH Zurich

   Affiliation as printed

   ETH Zurich , AI Center , Switzerland

5. Bastian Leibe Aachen

   RWTH Aachen University

   Affiliation as printed

   RWTH Aachen University , Germany