预测自动驾驶汽车(AV)周围的实体的行为是实现全自动驾驶的关键难题之一。使用我们的预测数据集和L5级工具包,您可以在一个闲暇的下午或周末开始构建行为预测模型,即使您之前没有任何有关自动驾驶汽车的经验。如果成功了,您可以帮助推动自动驾驶行业整体向前发展,如果您的解决方案在我们的Kaggle竞争中取得优胜,您甚至可以赢得我们30000美元奖金的一部分。
什么是行为预测么它这么重要strong>
尽管感知技术在自动驾驶行业已经发展得较为成熟了,但预测和规划仍需改进。那么为什么这两部分在自动驾驶堆栈中的问题还没解决呢p>
目前,行为预测和规划的模型主要是使用基于规则的系统建立的。然而,未来是不确定的,规则可能并不总是随不确定性而变化。当您将其他实体添加到组合中时,规则的数量和制定规则的成本将成倍增长。深度监督学习方法可以解决这一问题,但需要大量数据来正确捕捉道路上罕见和意外的行为。好消息是:我们的预测数据集是同类数据集中最大的,并且包含了这些罕见的行为。
2.为任务获取输入和输出
预测数据集以不同的时间戳记录自动驾驶汽车周围的世界。每个时间戳包括:
1. 帧
帧是自动驾驶汽车本身的记录。包含它的位置和旋转,以及在那一瞬间检测到的所有实体和红绿灯的列表。
2. 实体
实体是真实世界中可移动的实体。实体通过类别(汽车、行人等)和位置信息进行标记。实体也有唯一的身份,通过连续的帧进行追踪。
获取帧和实体的信息非常简单,只需运行以下代码片段:
3.定义模型
定义一个使用输入和输出进行训练的模型。您可以通过调整标准CNN架构(例如ResNet50)来创建一个简单的基线。尽管您可以保留原 络的核心部分,但您需要更改其输入和输出层以匹配您的设置。
为此,将第一个卷积层中的通道数与俯视图中的通道数匹配。一个3通道的卷积层不足以将不同层次的不同语义信息栅格化。下一步,确保输出数与您未来的预测范围(下面的示例中使用XY位移表示)相匹配。对于50步的视野,最后一层 络总共需要100个神经元。
4.训练模型
现在一切准备就绪,您可以根据数据训练基线模型了。将俯视图输入该模型,预测未来的轨迹。您可以设置均方误差(MSE)标准,并使用ADAM优化器优化 络。在Pythorch中,是下面这样:

我们也可以在几次迭代后,在以自动驾驶汽车为中心的俯视图光栅上绘制预测的轨迹。
进一步训练你的模型
以下是一些改进和测试模型的一些想法和小技巧:
l 如果你想提高速度并想用EfficientNet来取代ResNet,只需更换第一层和最后一层。
l 想看看实体的历史记录是否可以提高性能吗是一个改变一个配置值的问题,一切都是现成的。
l 如果光栅看起来太粗糙,可以提高其分辨率,甚至可以更改其纵横比。
l 每个实体仅一条轨迹不足以捕捉不确定性已经提供了多模型预测的评分标准,因此您可以直接了解您的模型是否有改进。
文章知识点与官方知识档案匹配,可进一步学习相关知识OpenCV技能树首页概览11507 人正在系统学习中 相关资源:典型自动驾驶仿真软件介绍-rtl8305nb-cg_datasheet-交通文档类…
声明:本站部分文章及图片源自用户投稿,如本站任何资料有侵权请您尽早请联系jinwei@zod.com.cn进行处理,非常感谢!