HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Embracing Single Stride 3D Object Detector with Sparse Transformer

Lue Fan Ziqi Pang Tianyuan Zhang Yu-Xiong Wang Hang Zhao Feng Wang Naiyan Wang Zhaoxiang Zhang

Embracing Single Stride 3D Object Detector with Sparse Transformer

Abstract

In LiDAR-based 3D object detection for autonomous driving, the ratio of the object size to input scene size is significantly smaller compared to 2D detection cases. Overlooking this difference, many 3D detectors directly follow the common practice of 2D detectors, which downsample the feature maps even after quantizing the point clouds. In this paper, we start by rethinking how such multi-stride stereotype affects the LiDAR-based 3D object detectors. Our experiments point out that the downsampling operations bring few advantages, and lead to inevitable information loss. To remedy this issue, we propose Single-stride Sparse Transformer (SST) to maintain the original resolution from the beginning to the end of the network. Armed with transformers, our method addresses the problem of insufficient receptive field in single-stride architectures. It also cooperates well with the sparsity of point clouds and naturally avoids expensive computation. Eventually, our SST achieves state-of-the-art results on the large scale Waymo Open Dataset. It is worth mentioning that our method can achieve exciting performance (83.8 LEVEL 1 AP on validation split) on small object (pedestrian) detection due to the characteristic of single stride. Codes will be released at https://github.com/TuSimple/SST

Code Repositories

tusen-ai/sst
pytorch
Mentioned in GitHub
tusimple/sst
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
3d-object-detection-on-waymo-cyclistSST
APH/L2: 72.17
3d-object-detection-on-waymo-pedestrianSST
APH/L2: 73.51
3d-object-detection-on-waymo-vehicleSST
APH/L2: 72.74

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp