自动驾驶预测与决策规划(nuplan数据集)

欢迎大家关注我的B站:

偷吃薯片的Zheng同学的个人空间-偷吃薯片的Zheng同学个人主页-哔哩哔哩视频 (bilibili.com)

目录

1.概述

[2 数据采集](#2 数据采集)

3.开环与闭环仿真

4.数据注释

5.场景

6.规划框架

6.1Train

6.2Simulation

6.3Metric

6.4Visualization

7.下载


本文参考nuPlan (nuscenes.org)

1.概述

nuPlan是全球首个大规模的自动驾驶规划的benchmark。虽然基于机器学习的运动规划算法越来越多,但由于缺乏成熟的数据集、模拟框架和指标,限制了这一领域的进展。

现有的自动驾驶汽车运动预测基准(ArgoverseLyftWaymo)侧重于其他智能体的短期运动预测,而不是自我车辆的长期规划 。这导致以前的工作使用开环评估 ,这不适合公平地评估长期规划。nuPlan提供了一个训练框架来开发一个learning based planner 、一个轻量级的closed-loop simulator ,、一个运动规划的specific metrics 和一个可视化结果的interactive tool

2 数据采集

nuplan提供了一个大型数据集,其中包含来自美国和亚洲 1200 个交通模式差异很大的城市(波士顿、匹兹堡、拉斯维加斯和新加坡)的 4 小时人类驾驶数据。

我们的目标是多样化的驾驶条件,但不包括大雨和夜间数据。所有数据都是通过手动驾驶收集的,以获得适合模仿学习的地面实况。我们的车辆操作员被指示安全、自信和自信地驾驶。为了使这些数据对规划有用,他们以目标为导向。这意味着他们定义一个目标(上下车点、公交车站、停车场),然后直接驶向目标。

我们使用以下传感器:

  • 5 倍激光雷达 (20Hz)
  • 2x Pandar 20P(前/后保险杠)
  • 2x Pandar 40P(左/右 A 柱)
  • 1x Pandar 40PT(车顶)
  • 8 倍摄像头 (10Hz)
  • D3 工程 D3RCM
  • Sony IMX390,2000 x 1200 分辨率,分割像素图像传感器
  • 1 个霍尼韦尔 HG1120 IMU (100Hz)
  • 1 个 Trimble BX992 (20Hz)

所有传感器都根据其外在和内在(仅限相机)进行校准。摄像头和激光雷达相互同步。

3.开环与闭环仿真

支持闭环和开环仿真。闭环意味着自我车辆和其他代理车辆可以偏离原始日志中记录的内容。除了为规划人员提供基线实施外,我们还为传统(智能驾驶员模型,IDM)和基于ML的智能代理提供baseline

  • 为了真实地模拟交通流量,nuplan开发了一种交通信号灯状态推断系统,该系统根据场景中观察到的车辆运动来推断交通信号灯的状态。最后,我们挖掘手工制作的场景(例如变道、行人与汽车的交互),以找到有趣的场景,并评估这些场景中的常见指标和特定场景的指标。

4.数据注释

与 nuScenes 类似,我们提供详细的 2D 高清地图,这些地图由人类使用语义类别进行注释,例如道路、人行道、人行横道、车道、交通信号灯等等。

我们输出以下类。包装盒标签由不同的颜色表示:

  • 车辆(橙色):包括所有四辆或更多轮式车辆以及拖车。
  • 自行车(红色):包括自行车、摩托车和三轮车。
  • 行人(蓝色):所有类型的行人,包括婴儿车和轮椅。
  • 交通锥(黑色):临时放置的控制交通流量的锥。
  • 屏障(黄色):可以是临时或永久的固体屏障。
  • 施工区标志(未显示):指示施工区的临时标志。
  • 通用物体(绿色):动物、碎片、可推/可拉物体、永久杆。

请注意,我们仅输出可行驶路面和人行道上被认为与自动驾驶汽车相关的选定部分的感知轨迹。其他轨道使用语义映射进行过滤。

5.场景

为了公平地评估运动规划模型,考虑基础数据分布至关重要。大多数时候,车辆都是直行的,因此规划是微不足道的。正是在具有挑战性的极端情况下,规划器才能真正展示其性能。为此,我们开发了一个框架,用于在整个数据集中挖掘和注释感兴趣的场景。要了解有关此场景挖掘框架的更多信息,请参阅这篇博文

下面我们展示了一些具有挑战性的场景示例,正如我们的系统所发现的那样:
无保护的横转弯


密集的车辆交互


前面的乱穿马路者


变道


在接送区自我

6.规划框架

nuPlan提供了一个开源工具包,用于开发基于ML的规划器。下面我们将介绍此工作流程中的四个步骤,即:训练 模型、模拟 模型、测量 性能和可视化结果。

6.1Train

我们开发了一个基于 PyTorch 的框架,使用许多输入来训练基于机器学习的规划人员,例如自我信息、其他代理信息、静态/动态地图信息。我们的基线可以很容易地修改和扩展,以纳入新的研究思路。

6.2Simulation

我们框架的核心部分是场景模拟。它被设计为在数据集、观察模型、自我规划器、代理模型、运动控制器、指标和可视化方面完全模块化。虽然我们为这些元素中的每一个都提供了多个选项和基线,但用户可以很容易地插入替换。这将使研究界能够专注于单个框架,而不是像以前那样对每个数据集和规划方法使用不同的代码。

6.3Metric

在一组方案上评估规划师后,我们计算了许多指标,这些指标分为 5 类:

  • 交通规则违规用于衡量对常见交通规则的遵守情况。我们计算与其他智能体的碰撞率、越野轨迹的发生率、与领先智能体的时间间隔、碰撞时间以及通过智能体时的相对速度作为通过距离的函数。
  • 人类驾驶相似性用于量化与人类相比的机动满意度,例如纵向速度误差、纵向停止位置误差和横向位置误差。此外,将产生的颠簸/加速度与人类水平的颠簸/加速度进行比较。
  • 车辆动力学量化了骑手的舒适度和轨迹的可行性。骑手舒适度是通过颠簸、加速度、转向率和车辆振荡来衡量的。可行性是通过违反相同标准的预定义限制来衡量的。
  • 目标实现使用 L2 距离衡量地图上目标航点的路线进度
  • 特定于方案的指标适用于特定方案。例如,对于变道,测量和评分目标车道上的碰撞时间和前方/后方代理的时间间隔。对于行人和骑自行车者的互动,我们量化了通过的相对速度,同时区分了他们的位置。

6.4Visualization

对于研究人员来说,要快速迭代新的 ML 规划模型,能够仔细分析模型的性能至关重要。为此,我们开发了带有基于散景的交互式模拟查看器的"nuBoard"仪表板,该查看器可以渲染底层语义图、自我车辆、场景中的其他智能体以及交通信号灯。除了场景模拟之外,我们还绘制了有关计划人员绩效的汇总统计量。

7.下载

这里我们列出了 nuPlan 数据集的不同版本。

免费下载nuPlan数据集用于非商业用途,以便我们更好地跟踪其使用情况。

此数据集也可在 AWS 上的 Registry of Open Data 上获得。

可以在 GitHub 存储库中找到 nuPlan 数据集的 devkit。

相关推荐
搏博21 分钟前
神经网络问题之一:梯度消失(Vanishing Gradient)
人工智能·机器学习
z千鑫21 分钟前
【人工智能】深入理解PyTorch:从0开始完整教程!全文注解
人工智能·pytorch·python·gpt·深度学习·ai编程
YRr YRr30 分钟前
深度学习:神经网络的搭建
人工智能·深度学习·神经网络
威桑32 分钟前
CMake + mingw + opencv
人工智能·opencv·计算机视觉
爱喝热水的呀哈喽36 分钟前
torch张量与函数表达式写法
人工智能·pytorch·深度学习
EAI-Robotics1 小时前
机器人打包物品研究现状简述
机器人
肥猪猪爸1 小时前
使用卡尔曼滤波器估计pybullet中的机器人位置
数据结构·人工智能·python·算法·机器人·卡尔曼滤波·pybullet
LZXCyrus2 小时前
【杂记】vLLM如何指定GPU单卡/多卡离线推理
人工智能·经验分享·python·深度学习·语言模型·llm·vllm
我感觉。2 小时前
【机器学习chp4】特征工程
人工智能·机器学习·主成分分析·特征工程
YRr YRr2 小时前
深度学习神经网络中的优化器的使用
人工智能·深度学习·神经网络