用小波去噪提升事件相机实时目标检测性能
WD-DETR: Wavelet Denoising-Enhanced Real-Time Object Detection Transformer for Robot Perception with Event Cameras
- 先将事件数据转为张量,再用小波变换滤除噪声
- 在DSEC、Gen1、1Mpx数据集上均超越现有方法
- 可在Jetson Orin NX实现35 FPS,适合机器人实时感知
以往基于事件相机的研究使用密集事件表示实现了不错的检测性能,但累积噪声对表示质量的影响未受足够重视,导致漏检率上升。为此,本文提出波形去噪增强型检测变压器(WD-DETR),用于事件相机。首先构建一种密集事件表示,实现实时事件张量重建;随后设计小波变换方法过滤事件表示中的噪声,并将其集成到主干网络中进行特征提取;提取的特征输入基于Transformer的网络完成目标预测。为进一步降低推理时间,引入动态重组卷积块(DRCB)作为混合编码器中的融合模块。该方法在三个事件相机目标检测数据集(DSEC、Gen1、1Mpx)上评估,结果表明其优于当前主流方法。此外,在机器人常用嵌入式平台NVIDIA Jetson Orin NX上部署,结合TensorRT FP16实现约35 FPS的高帧率,非常适用于机器人系统的实时感知。
原文摘要 · Abstract (English)
Previous studies on event camera sensing have demonstrated certain detection performance using dense event representations. However, the accumulated noise in such dense representations has received insufficient attention, which degrades the representation quality and increases the likelihood of missed detections. To address this challenge, we propose the Wavelet Denoising-enhanced DEtection TRansformer, i.e., WD-DETR network, for event cameras. In particular, a dense event representation is presented first, which enables real-time reconstruction of events as tensors. Then, a wavelet transform method is designed to filter noise in the event representations. Such a method is integrated into the backbone for feature extraction. The extracted features are subsequently fed into a transformer-based network for object prediction. To further reduce inference time, we incorporate the Dynamic Reorganization Convolution Block (DRCB) as a fusion module within the hybrid encoder. The proposed method has been evaluated on three event-based object detection datasets, i.e., DSEC, Gen1, and 1Mpx. The results demonstrate that WD-DETR outperforms tested state-of-the-art methods. Additionally, we implement our approach on a common onboard computer for robots, the NVIDIA Jetson Orin NX, achieving a high frame rate of approximately 35 FPS using TensorRT FP16, which is exceptionally well-suited for real-time perception of onboard robotic systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。