Shihao Wang, Yingfei Liu, Tiancai Wang, Ying Li, Xiangyu Zhang
In this paper, we propose a long-sequence modeling framework, named StreamPETR, for multi-view 3D object detection. Built upon the sparse query design in the PETR series, we systematically develop an object-centric temporal mechanism. The model is performed in an online manner and the long-term historical information is propagated through object queries frame by frame. Besides, we introduce a motion-aware layer normalization to model the movement of the objects. StreamPETR achieves significant performance improvements only with negligible computation cost, compared to the single-frame baseline. On the standard nuScenes benchmark, it is the first online multi-view method that achieves comparable performance (67.6% NDS & 65.3% AMOTA) with lidar-based methods. The lightweight version realizes 45.0% mAP and 31.7 FPS, outperforming the state-of-the-art method (SOLOFusion) by 2.3% mAP and 1.8x faster FPS. Code has been available at https://github.com/exiawsh/StreamPETR.git.
| Task | Dataset | Metric | Value | Model |
|---|---|---|---|---|
| Multi-Object Tracking | nuScenes Camera Only | AMOTA | 65.3 | StreamPETR-Large |
| Object Tracking | nuScenes Camera Only | AMOTA | 65.3 | StreamPETR-Large |
| Object Detection | nuScenes Camera Only | NDS | 67.6 | StreamPETR-Large |
| Object Detection | 3D Object Detection on Argoverse2 Camera Only | Average mAP | 20.3 | StreamPETR |
| 3D | nuScenes Camera Only | NDS | 67.6 | StreamPETR-Large |
| 3D | 3D Object Detection on Argoverse2 Camera Only | Average mAP | 20.3 | StreamPETR |
| 3D Object Detection | nuScenes Camera Only | NDS | 67.6 | StreamPETR-Large |
| 3D Object Detection | 3D Object Detection on Argoverse2 Camera Only | Average mAP | 20.3 | StreamPETR |
| 3D Multi-Object Tracking | nuScenes Camera Only | AMOTA | 65.3 | StreamPETR-Large |
| 2D Classification | nuScenes Camera Only | NDS | 67.6 | StreamPETR-Large |
| 2D Classification | 3D Object Detection on Argoverse2 Camera Only | Average mAP | 20.3 | StreamPETR |
| 2D Object Detection | nuScenes Camera Only | NDS | 67.6 | StreamPETR-Large |
| 2D Object Detection | 3D Object Detection on Argoverse2 Camera Only | Average mAP | 20.3 | StreamPETR |
| 16k | nuScenes Camera Only | NDS | 67.6 | StreamPETR-Large |
| 16k | 3D Object Detection on Argoverse2 Camera Only | Average mAP | 20.3 | StreamPETR |