阿里达摩院突破自动驾驶技术难题:实现3D物体检测精度速度兼得

新闻 无人驾驶
达摩院的研究人员提出一个通用、高性能的检测器,在自动驾驶领域最重要的测试集之一 KITTI的鸟瞰(BEV)数据集上,检测速度达到25FPS ,一举占据榜首

 本文经AI新媒体量子位(公众号ID:QbitAI)授权转载,转载请联系出处。

最新消息,阿里达摩院又有新的研究进展,这次在自动驾驶3D物体检测领域

达摩院的研究人员提出一个通用、高性能的检测器,在自动驾驶领域最重要的测试集之一 KITTI的鸟瞰(BEV)数据集上,检测速度达到25FPS ,一举占据榜首,与排名第二的方案相比,减少了一半多,同时精度也远超其他的单阶段检测器。可以说首次实现3D物体检测精度与速度兼得。

阿里达摩院突破自动驾驶技术难题:实现3D物体检测精度速度兼得

达摩院研究团队表示:“检测器是自动驾驶系统的核心组件之一,但该领域一直缺少创新和突破,此次我们实现3D检测精读和速度的提升,将有助于提高自动驾驶系统的安全性。”

从实验结果来看,他们显然取得了良好进展,而且研究的主要完成人员(一作),还是一名达摩院的实习生。

同时,该研究也得到了同行们的认可,被计算机视觉领域顶级会议CVPR 2020收录。他们是如何做到的呢?我们借助阿里达摩院提供的解读,一一来看。

如何实现精度和速度兼得?

众所周知,与普通2D图像识别应用不同,自动驾驶系统对精度和速度的要求更高,不仅需要快速识别周围环境的物体,还要对物体在三维空间中的位置做精准定位。

仅靠传感器和算法模型,通常无法平衡视觉识别的精度和速度。因此,检测器成为提升自动驾驶系统安全性的一个关键因素。

阿里达摩院突破自动驾驶技术难题:实现3D物体检测精度速度兼得

经过多年研究,当前业界主流的单阶段检测器在检测速度上很好,但在检测精度却差强人意。

这就是达摩院研究的出发点:寻找一种能二者兼得的方法。

他们提出的思路是:将两阶段检测器中对特征进行细粒度刻画的思想,移植到单阶段检测中。

在他们的模型中,用于部署的检测器, 即推断网络, 由一个骨干网络和检测头组成。

骨干网络用3D的稀疏网络实现,用于提取含有高语义的体素特征。检测头将体素特征压缩成鸟瞰图表示,并在上面运行2D全卷积网络来预测3D物体框。

他们在训练中利用一个辅助网络将单阶段检测器中的体素特征,转化为点级特征并施加一定的监督信号。

在实现上,他们将卷积特征中的非零信号映射到原始的点云空间中, 然后在每个点上进行插值,来获取卷积特征的点级表示。使得卷积特征也具有结构感知能力,来提高检测精度。

阿里达摩院突破自动驾驶技术难题:实现3D物体检测精度速度兼得

而在做模型推断时,辅助网络并不参与计算(detached),保证单阶段检测器的检测效率。

另外,他们还提出一个工程上的改进:Part-sensitive Warping (PSWarp),用于处理单阶段检测器中存在的 “框-置信度-不匹配” 问题。

核心思路是:利用采样器, 用生成的采样网格在对应的局部敏感特征图上进行采样,生成对齐好的特征图。最终能反映置信度的特征图,是K个对齐好特征图的平均。

阿里达摩院突破自动驾驶技术难题:实现3D物体检测精度速度兼得

单阶段方法,能达到两阶段方法精度

阿里达摩院的研究人员,在KITTI数据集上评估了方法的有效性。下图(PR Curve)中,实线为两阶段方法, 虚线为单阶段方法。

阿里达摩院突破自动驾驶技术难题:实现3D物体检测精度速度兼得

可以看出,达摩院提出的单阶段方法(黑色)能够达到两阶段方法才能达到的精度。

下图展示了他们在KITTI 鸟瞰(BEV) 和 3D 测试集上的结果。

阿里达摩院突破自动驾驶技术难题:实现3D物体检测精度速度兼得

可以看出,他们提出的方法,可以在不增加额外计算量的情况下,达到25FPS 的检测速度,而且还能保持精度。具体的检测效果如下:

阿里达摩院突破自动驾驶技术难题:实现3D物体检测精度速度兼得

两位共同一作,都是达摩院研究实习生

研究论文,标题为“Structure Aware Single-Stage 3D Object Detection from Point Cloud”,共有5名研究人员参与,分别来自阿里达摩院和香港理工大学。

第一作者是Chenhang He,是阿里达摩院的研究实习生,正在香港理工大学读博,预计2022年毕业。

他的导师是达摩院高级研究员、香港理工大学电子计算学系讲座教授、IEEE Fellow张磊,也是这一研究的通讯作者。

另一位第一作者Hui Zeng,也是是阿里达摩院的研究实习生,同样是张磊的博士生,预计在今年毕业。

其他作者,分别是达摩院高级研究员、IEEE Fellow华先胜、达摩院资深算法专家黄建强等。

责任编辑:张燕妮 来源: 量子位
相关推荐

2023-11-17 09:55:41

2023-06-02 10:33:35

2024-01-26 10:02:51

自动驾驶3D

2018-05-14 08:55:58

自动驾驶数据人工智能

2022-07-13 10:20:14

自动驾驶3D算法

2020-08-13 10:41:14

达摩院阿里自动驾驶

2023-12-12 10:09:33

3D框架模型

2022-01-20 11:17:27

自动驾驶智能汽车

2022-01-06 09:41:28

自动驾驶数据人工智能

2022-02-14 11:37:59

自动驾驶算法技术

2021-11-10 10:05:47

特斯拉自动驾驶技术

2023-12-12 10:23:54

自动驾驶雷达视觉

2024-12-04 10:00:00

自动驾驶3D检测

2020-08-26 10:37:21

阿里3D

2024-08-27 10:20:00

2022-06-23 12:12:31

达摩院自动驾驶

2021-12-01 10:21:27

自动驾驶技术人工智能

2018-11-14 18:00:18

自动驾驶安全人工智能

2022-02-17 10:22:17

汽车智能自动驾驶

2023-03-15 11:54:32

无人驾驶系统
点赞
收藏

51CTO技术栈公众号