arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.09751cs.CV

TASeg: 面向激光雷达语义分割的时序聚合网络

TASeg: Temporal Aggregation Network for LiDAR Semantic Segmentation

Zhejiang University(浙江大学);Shanghai AI Laboratory(上海人工智能实验室);Fullong Inc.(福隆公司);School of Software Technology, Zhejiang University(浙江大学软件学院);The Chinese University of Hong Kong(香港中文大学);ShanghaiTech University(上海科技大学)

Xiaopei Wu, Yuenan Hou, Xiaoshui Huang, Binbin Lin, Tong He, Xinge Zhu, Yuexin Ma, Boxi Wu, Haifeng Liu, Deng Cai, Wanli Ouyang

AI总结:

针对激光雷达点云稀疏导致语义分割模型训练难的问题,提出TASeg网络,通过TLAD算法、TIAF模块和SMSA增强充分挖掘时序多模态信息,在三个主流基准赛道上取得第一。

摘要:

由于点云固有的稀疏性,训练用于激光雷达语义分割的深度模型颇具挑战。利用时序数据是解决稀疏性问题的天然方案,因为它能让输入信号更密集。然而,以往的多帧融合算法受内存限制,无法充分利用足够的时序信息,同时也忽略了富含信息的时序图像。为充分挖掘长期时序点云和图像中隐藏的丰富信息,我们提出了时序聚合网络,命名为TASeg。具体而言,我们提出了时序激光雷达聚合与蒸馏(Temporal LiDAR Aggregation and Distillation, TLAD)算法,该算法利用历史先验为不同类别分配不同的聚合步数,可在大幅降低内存和时间开销的同时实现更高的精度。此外,TLAD训练一个注入了真值先验的教师模型来蒸馏目标模型,进一步提升性能。为充分利用时序图像,我们设计了时序图像聚合与融合(Temporal Image Aggregation and Fusion, TIAF)模块,该模块能大幅扩展相机视场(FOV)并增强当前帧特征。相机视场内的时序激光雷达点被作为媒介,将时序图像特征转换到当前坐标系下,以实现时序多模态融合。此外,我们开发了静态-动态切换增强(Static-Moving Switch Augmentation, SMSA)算法,该算法利用充足的时序信息让物体能够自由切换运动状态,从而大幅增加静态和动态训练样本的数量。我们的TASeg在三个具有挑战性的赛道上排名第一,即SemanticKITTI单扫描赛道、多扫描赛道以及nuScenes激光雷达分割赛道,有力证明了我们方法的优越性。代码可在https://github.com/LittlePey/TASeg获取。

英文摘要:

Training deep models for LiDAR semantic segmentation is challenging due to the inherent sparsity of point clouds. Utilizing temporal data is a natural remedy against the sparsity problem as it makes the input signal denser. However, previous multi-frame fusion algorithms fall short in utilizing sufficient temporal information due to the memory constraint, and they also ignore the informative temporal images. To fully exploit rich information hidden in long-term temporal point clouds and images, we present the Temporal Aggregation Network, termed TASeg. Specifically, we propose a Temporal LiDAR Aggregation and Distillation (TLAD) algorithm, which leverages historical priors to assign different aggregation steps for different classes. It can largely reduce memory and time overhead while achieving higher accuracy. Besides, TLAD trains a teacher injected with gt priors to distill the model, further boosting the performance. To make full use of temporal images, we design a Temporal Image Aggregation and Fusion (TIAF) module, which can greatly expand the camera FOV and enhance the present features. Temporal LiDAR points in the camera FOV are used as mediums to transform temporal image features to the present coordinate for temporal multi-modal fusion. Moreover, we develop a Static-Moving Switch Augmentation (SMSA) algorithm, which utilizes sufficient temporal information to enable objects to switch their motion states freely, thus greatly increasing static and moving training samples. Our TASeg ranks 1st on three challenging tracks, i.e., SemanticKITTI single-scan track, multi-scan track and nuScenes LiDAR segmentation track, strongly demonstrating the superiority of our method. Codes are available at https://github.com/LittlePey/TASeg.

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

↑