Recurrent Vision Transformers for Object Detection with Event Cameras

被引:38
作者
Gehrig, Mathias [1 ]
Scaramuzza, Davide [1 ]
机构
[1] Univ Zurich, Robot & Percept Grp, Zurich, Switzerland
来源
2023 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR) | 2023年
基金
瑞士国家科学基金会; 欧洲研究理事会;
关键词
D O I
10.1109/CVPR52729.2023.01334
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
We present Recurrent Vision Transformers (RVTs), a novel backbone for object detection with event cameras. Event cameras provide visual information with sub-millisecond latency at a high-dynamic range and with strong robustness against motion blur. These unique properties offer great potential for low-latency object detection and tracking in time-critical scenarios. Prior work in event-based vision has achieved outstanding detection performance but at the cost of substantial inference time, typically beyond 40 milliseconds. By revisiting the high-level design of recurrent vision backbones, we reduce inference time by a factor of 6 while retaining similar performance. To achieve this, we explore a multi-stage design that utilizes three key concepts in each stage: first, a convolutional prior that can be regarded as a conditional positional embedding. Second, local and dilated global self-attention for spatial feature interaction. Third, recurrent temporal feature aggregation to minimize latency while retaining temporal information. RVTs can be trained from scratch to reach state-of-the-art performance on event-based object detection - achieving an mAP of 47.2% on the Gen1 automotive dataset. At the same time, RVTs offer fast inference (< 12 ms on a T4 GPU) and favorable parameter efficiency (5x fewer than prior art). Our study brings new insights into effective design choices that can be fruitful for research beyond event-based vision. Code: https://github.com/uzh-rpg/RVT
引用
收藏
页码:13884 / 13893
页数:10
相关论文
共 58 条
  • [1] Arnab Anurag, 2021, INT C COMP VIS ICCV
  • [2] Ba J. Lei, 2016, ARXIV E PRINTS
  • [3] Baldwin R., 2022, IEEE Transactions on Pattern Analysis and Machine Intelligence
  • [4] Barchid Sami, 2022, IEEE INT C IM PROC I
  • [5] Bellec G., 2020, NATURE COMMUNICATION
  • [6] Bertasius G, 2021, PR MACH LEARN RES, V139
  • [7] Cannici Marco, 2019, IEEE C COMP VIS PATT
  • [8] Chen Nicholas F. Y., 2018, IEEE C COMP VIS PATT
  • [9] Chu XX, 2021, ADV NEUR IN
  • [10] Chu Xiangxiang, 2021, Conditional Positional Encodings for Vision Transformers