4 个月前

RRPN:用于自动驾驶车辆中目标检测的雷达区域提议网络

RRPN:用于自动驾驶车辆中目标检测的雷达区域提议网络

摘要

区域提议算法在大多数最先进的两阶段目标检测网络中起着重要作用,通过假设图像中的目标位置来生成候选区域。然而,区域提议算法被认为是大多数两阶段目标检测网络的瓶颈,增加了每张图像的处理时间,导致这些网络无法满足自动驾驶车辆等实时应用的需求。本文介绍了一种基于雷达的实时区域提议算法(RRPN),用于自动驾驶车辆中的目标检测。RRPN通过将雷达检测结果映射到图像坐标系,并为每个映射的雷达检测点生成预定义的锚框来生成目标提议。然后根据目标与车辆的距离对这些锚框进行变换和缩放,以提供更精确的目标提议。我们在新发布的NuScenes数据集[1]上使用Fast R-CNN目标检测网络[2]评估了我们的方法。与选择性搜索目标提议算法[3]相比,我们的模型运行速度提高了100倍以上,同时实现了更高的检测精度和召回率。代码已公开发布在https://github.com/mrnabati/RRPN 。

代码仓库

mrnabati/RRPN
官方
caffe2
GitHub 中提及

基准测试

基准方法指标
3d-object-detection-on-nuscenes-fRRPN + R101 - F
AP: 43
AP50: 64.9
AP75: 48.5
AR: 48.6
ARI: 58.2
ARm: 41.2
ARs: 4
3d-object-detection-on-nuscenes-fbRRPN + R101 - FB
AP: 35.5
AP50: 59
AP75: 37
AR: 42.1
ARI: 51.4
ARm: 39.1
ARs: 21.1

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
RRPN:用于自动驾驶车辆中目标检测的雷达区域提议网络 | 论文 | HyperAI超神经