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

LidarMultiNet:面向LiDAR感知的统一多任务网络

LidarMultiNet: Towards a Unified Multi-Task Network for LiDAR Perception

TuSimple(图森未来)

Dongqiangzi Ye, Zixiang Zhou, Weijia Chen, Yufei Xie, Yu Wang, Panqu Wang, Hassan Foroosh

摘要:

基于LiDAR的3D对象检测、语义分割和全景分割通常在具有难以相互适应的独特架构的专用网络中实现。本文介绍了LidarMultiNet,这是一个基于LiDAR的多任务网络,它统一了这三个主要的LiDAR感知任务。多任务网络有许多优点,其中之一是可以通过在多个任务之间共享权重和计算来降低总体成本。然而,与独立组合的单任务模型相比,它通常表现不佳。LidarMultiNet旨在弥合多任务网络和多个单任务网络之间的性能差距。LidarMultiNet的核心是一个强大的基于3D体素的编码器-解码器架构,具有全局上下文池(GCP)模块,可从LiDAR帧中提取全局上下文特征。任务特定的头被添加到网络的顶部,以执行三个激光雷达感知任务。只需添加新的任务专用头,即可实现更多任务,同时几乎不增加成本。还提出了第二阶段以细化第一阶段分割并生成准确的全景分割结果。LidarMultiNet在Waymo Open Dataset和nuScenes数据集上进行了广泛的测试,首次证明主要的LiDAR感知任务可以统一在一个强大的网络中,该网络经过端到端的训练,并实现了最先进的性能。值得注意的是,LidarMultiNet在2022年Waymo开放数据集3D语义分割挑战赛中获得了官方第一名,在测试集中的22个类别中,仅使用LiDAR点作为输入,mIoU最高,准确率最高。它还在Waymo 3D目标检测基准和三个nuScenes基准上为单个模型设置了新的最先进技术。

英文摘要:

LiDAR-based 3D object detection, semantic segmentation, and panoptic segmentation are usually implemented in specialized networks with distinctive architectures that are difficult to adapt to each other. This paper presents LidarMultiNet, a LiDAR-based multi-task network that unifies these three major LiDAR perception tasks. Among its many benefits, a multi-task network can reduce the overall cost by sharing weights and computation among multiple tasks. However, it typically underperforms compared to independently combined single-task models. The proposed LidarMultiNet aims to bridge the performance gap between the multi-task network and multiple single-task networks. At the core of LidarMultiNet is a strong 3D voxel-based encoder-decoder architecture with a Global Context Pooling (GCP) module extracting global contextual features from a LiDAR frame. Task-specific heads are added on top of the network to perform the three LiDAR perception tasks. More tasks can be implemented simply by adding new task-specific heads while introducing little additional cost. A second stage is also proposed to refine the first-stage segmentation and generate accurate panoptic segmentation results. LidarMultiNet is extensively tested on both Waymo Open Dataset and nuScenes dataset, demonstrating for the first time that major LiDAR perception tasks can be unified in a single strong network that is trained end-to-end and achieves state-of-the-art performance. Notably, LidarMultiNet reaches the official 1st place in the Waymo Open Dataset 3D semantic segmentation challenge 2022 with the highest mIoU and the best accuracy for most of the 22 classes on the test set, using only LiDAR points as input. It also sets the new state-of-the-art for a single model on the Waymo 3D object detection benchmark and three nuScenes benchmarks.

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

↑