arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2410.11228cs.CV

TEOcc:基于时间增强的雷达-相机多模态占用预测

TEOcc: Radar-camera Multi-modal Occupancy Prediction via Temporal Enhancement

Wangxuan Institute of Computer Technology, Peking University(北京大学王选计算机研究所);Chongqing Changan Automobile Co., Ltd.(重庆长安汽车股份有限公司)

Zhiwei Lin, Hongbo Jin, Yongtao Wang, Yufei Wei, Nan Dong

AI总结:

TEOcc提出一种雷达-相机多模态时间增强占用预测网络,通过引入仅在训练时使用的时间增强分支学习长短期时间信息,并在nuScenes基准上实现了最先进的占用预测性能。

摘要:

作为一种新颖的3D场景表示,语义占用在自动驾驶中受到了广泛关注。然而,现有的占用预测方法主要集中于设计更好的占用表示,如三视角图或神经辐射场,却忽略了使用长时间序列信息的优势。在本文中,我们提出了一种雷达-相机多模态时间增强占用预测网络,简称TEOcc。我们的方法受到3D目标检测中利用时间信息取得成功的启发。具体而言,我们引入了一个时间增强分支来学习时间占用预测。在该分支中,我们随机丢弃多视角相机的t-k输入帧,并分别利用长期和短期时间解码器,结合来自其他相邻帧和多模态输入的信息来预测其3D占用。此外,为了降低计算成本并融合多模态输入,我们专门为长期和短期时间解码器设计了3D卷积层。进一步地,由于轻量级的占用预测头是一个密集分类头,我们提出在时间增强分支和主分支之间使用共享的占用预测头。值得注意的是,时间增强分支仅在训练期间执行,在推理期间被丢弃。实验结果表明,TEOcc在nuScenes基准测试上实现了最先进的占用预测。此外,所提出的时间增强分支是一个即插即用模块,可以轻松集成到现有的占用预测方法中以提升占用预测的性能。代码和模型将发布于https://github.com/VDIGPKU/TEOcc。

英文摘要:

As a novel 3D scene representation, semantic occupancy has gained much attention in autonomous driving. However, existing occupancy prediction methods mainly focus on designing better occupancy representations, such as tri-perspective view or neural radiance fields, while ignoring the advantages of using long-temporal information. In this paper, we propose a radar-camera multi-modal temporal enhanced occupancy prediction network, dubbed TEOcc. Our method is inspired by the success of utilizing temporal information in 3D object detection. Specifically, we introduce a temporal enhancement branch to learn temporal occupancy prediction. In this branch, we randomly discard the t-k input frame of the multi-view camera and predict its 3D occupancy by long-term and short-term temporal decoders separately with the information from other adjacent frames and multi-modal inputs. Besides, to reduce computational costs and incorporate multi-modal inputs, we specially designed 3D convolutional layers for long-term and short-term temporal decoders. Furthermore, since the lightweight occupancy prediction head is a dense classification head, we propose to use a shared occupancy prediction head for the temporal enhancement and main branches. It is worth noting that the temporal enhancement branch is only performed during training and is discarded during inference. Experiment results demonstrate that TEOcc achieves state-of-the-art occupancy prediction on nuScenes benchmarks. In addition, the proposed temporal enhancement branch is a plug-and-play module that can be easily integrated into existing occupancy prediction methods to improve the performance of occupancy prediction. The code and models will be released at https://github.com/VDIGPKU/TEOcc.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑