CenterNet3D: An Anchor free Object Detector for Autonomous Driving




mmdetection目标检测手机使用检测 目标检测软件_3D


作者团队:吉林大学&中科院&中山大学等
论文下载链接: https:// arxiv.org/abs/2007.0721 4

注1:如果上述链接无法访问或者下载速度过慢,可以看文末,已上传至百度云,方便下载

注2:Anchor-free火于目标检测(19年初到现在),而现在Anchor-free + x的组合研究越来越多了,如应用到目标跟踪、3D目标检测等

CenterNet3D:一阶段、anchor-free、无NMS的3D目标检测新网络,性能优于PointPillars、TANet和SECOND等网络,速度可达26 FPS!

CenterNet3D

从点云进行准确,快速的3D目标检测是自动驾驶的关键任务。现有的一阶段3D目标检测方法可以实现实时性能,但是,它们被基于anchor的检测器所支配,该检测器效率低下并且需要额外的后处理。

在本文中,我们消除了anchor,并将对象建模为其边界框中心点的单个点。基于中心点,我们提出了anchor-free的CenterNet3D网络,该网络无需anchhor即可执行3D目标检测。

我们的CenterNet3D使用关键点估计来找到中心点并直接使3D边界框回归。但是,由于点云的固有稀疏性,因此3D对象中心点很可能位于空白区域,这使得难以估计准确的边界。

为了解决这个问题,我们提出了一个辅助角点注意力模块,以强制CNN backbone更加关注对象边界,这对于获得更准确的边界框是有效的。此外,我们的CenterNet3D不受非极大值抑制的影响,这使其更有效,更简单。


mmdetection目标检测手机使用检测 目标检测软件_tensorflow2 目标检测_02


mmdetection目标检测手机使用检测 目标检测软件_搜索_03


mmdetection目标检测手机使用检测 目标检测软件_3D_04


实验结果

在KITTI基准上,我们提出的CenterNet3D与其他基于anchor的一阶段方法取得了竞争性性能,这些方法显示了我们提出的中心点表征的有效性。


mmdetection目标检测手机使用检测 目标检测软件_搜索_05


mmdetection目标检测手机使用检测 目标检测软件_3D_06


mmdetection目标检测手机使用检测 目标检测软件_3D_07