用于自动驾驶的场景编码模型训练方法、装置及介质制造方法及图纸

本公开提供了一种用于自动驾驶的场景编码模型训练方法、装置及介质，涉及计算机技术领域，尤其涉及人工智能、自动驾驶领域。实现方案为：获取样本数据集；基于至少一个掩码策略，对多个第一样本数据进行处理，以获取多个第二样本数据；将多个第二样本数据输入场景编码模型，以获取场景编码模型输出的分别对应于多个第二样本数据的多个行驶场景编码；以及基于多个行驶场景编码训练场景编码模型。个行驶场景编码训练场景编码模型。个行驶场景编码训练场景编码模型。

【技术实现步骤摘要】
用于自动驾驶的场景编码模型训练方法、装置及介质


[0001]本公开涉及计算机
，尤其涉及人工智能、自动驾驶领域，具体涉及一种用于自动驾驶的场景编码模型训练方法、用于自动驾驶的场景编码方法、装置、电子设备、计算机可读存储介质和计算机程序产品。

技术介绍

[0002]人工智能是研究使计算机来模拟人的某些思维过程和智能行为(如学习、推理、思考、规划等)的学科，既有硬件层面的技术也有软件层面的技术。人工智能硬件技术一般包括如传感器、专用人工智能芯片、云计算、分布式存储、大数据处理等技术；人工智能软件技术主要包括计算机视觉技术、语音识别技术、自然语言处理技术以及机器学习/深度学习、大数据处理技术、知识图谱技术等几大方向。
[0003]基于人工智能的数据处理，已经广泛应用于各个领域。其中，在自动驾驶领域中，基于人工智能对数据进行处理，能够为车辆规划合理的驾驶轨迹。自动驾驶车辆依靠人工智能、视觉计算、雷达、监控装置和全球定位系统等协同合作，不需要人现场实际操控车辆便可以实现车辆的驾驶，是未来智能交通的主要发展方向之一。
[0004]在此部分中描述的方法不一定是之前已经设想到或采用的方法。除非另有指明，否则不应假定此部分中描述的任何方法仅因其包括在此部分中就被认为是现有技术。类似地，除非另有指明，否则此部分中提及的问题不应认为在任何现有技术中已被公认。

技术实现思路

[0005]本公开提供了一种用于自动驾驶的场景编码模型训练方法、用于自动驾驶的场景编码方法、装置、电子设备、计算机可读存储介质和计算机程序产品。
[0006]根据本公开的一方面，提供了一种用于自动驾驶的场景编码模型训练方法，包括：获取样本数据集，样本数据集包括对应于多个第一时刻的多个第一样本数据，每个第一样本数据包括该第一样本数据相应的第一时刻对应的第一数据以及相应第一时刻之前的至少一个历史时刻对应的至少一个第二数据，第一数据包括相应第一时刻的车辆行驶状态数据，至少一个第二数据中的每个第二数据包括相应历史时刻的车辆行驶状态数据；基于至少一个掩码策略，对多个第一样本数据进行处理，以获取多个第二样本数据，至少一个掩码策略包括第一掩码策略，第一掩码策略用于对多个第一样本数据中的至少一个第三样本数据中每个第三样本数据的第一数据进行掩码处理；将多个第二样本数据输入场景编码模型，以获取场景编码模型输出的分别对应于多个第二样本数据的多个行驶场景编码；以及基于多个行驶场景编码训练场景编码模型。
[0007]根据本公开的另一方面，提供了一种用于自动驾驶的场景编码方法，包括：获取当前时刻的第一数据以及当前时刻之前的至少一个历史时刻对应的至少一个第二数据，第一数据包括当前时刻的车辆行驶状态数据、障碍物状态数据和路网数据中的至少一者，至少一个第二数据中的每个第二数据包括相应历史时刻的车辆行驶状态数据、障碍物状态数据
和路网数据中的至少一者；将第一数据和至少一个第二数据输入场景编码模型中，以获得场景编码模型输出的当前时刻的行驶场景编码，其中，场景编码模型根据上述用于自动驾驶的场景编码模型训练方法训练获得。
[0008]根据本公开的另一方面，提供了一种用于自动驾驶的场景编码模型训练装置，包括：第一获取单元，被配置为获取样本数据集，样本数据集包括对应于多个第一时刻的多个第一样本数据，每个第一样本数据包括该第一样本数据相应的第一时刻对应的第一数据以及相应第一时刻之前的至少一个历史时刻对应的至少一个第二数据，第一数据包括相应第一时刻的车辆行驶状态数据，至少一个第二数据中的每个第二数据包括相应历史时刻的车辆行驶状态数据；处理单元，被配置为基于至少一个掩码策略，对多个第一样本数据进行处理，以获取多个第二样本数据，至少一个掩码策略包括第一掩码策略，第一掩码策略用于对多个第一样本数据中的至少一个第三样本数据中每个第三样本数据的第一数据进行掩码处理；输入单元，被配置为将多个第二样本数据输入场景编码模型，以获取场景编码模型输出的分别对应于多个第二样本数据的多个行驶场景编码；以及训练单元，被配置为基于多个行驶场景编码训练场景编码模型。
[0009]根据本公开的另一方面，提供了一种用于自动驾驶的场景编码装置，包括：第二获取单元，被配置为获取当前时刻的第一数据以及当前时刻之前的至少一个历史时刻对应的至少一个第二数据，第一数据包括当前时刻的车辆行驶状态数据、障碍物状态数据和路网数据中的至少一者，至少一个第二数据中的每个第二数据包括相应历史时刻的车辆行驶状态数据、障碍物状态数据和路网数据中的至少一者；编码单元，被配置为将第一数据和至少一个第二数据输入场景编码模型中，以获得场景编码模型输出的当前时刻的行驶场景编码，其中，场景编码模型根据上述用于自动驾驶的场景编码模型训练方法训练获得。
[0010]根据本公开的另一方面，提供了一种电子设备，包括：至少一个处理器；以及与所述至少一个处理器通信连接的存储器；其中所述存储器存储有可被所述至少一个处理器执行的指令，所述指令被所述至少一个处理器执行，以使所述至少一个处理器能够执行上述用于自动驾驶的场景编码模型训练方法或上述用于自动驾驶的场景编码方法。
[0011]根据本公开的另一方面，提供了一种存储有计算机指令的非瞬时计算机可读存储介质，其中，所述计算机指令用于使所述计算机执行上述用于自动驾驶的场景编码模型训练方法或上述用于自动驾驶的场景编码方法。
[0012]根据本公开的另一方面，提供了一种计算机程序产品，包括计算机程序，其中，所述计算机程序在被处理器执行时实现上述用于自动驾驶的场景编码模型训练方法或上述用于自动驾驶的场景编码方法。
[0013]根据本公开的一个或多个实施例，可以通过至少一种掩码策略，对多个第一样本数据中的一个或多个样本数据中的部分信息进行随机掩码(例如，对样本数据中的第一数据进行随机掩码处理)，并基于处理后的样本数据进行模型训练，从而提升模型的鲁棒性，使其能够在接收到第一数据异常的噪声数据时，有更准确的场景编码表达。
[0014]应当理解，本部分所描述的内容并非旨在标识本公开的实施例的关键或重要特征，也不用于限制本公开的范围。本公开的其它特征将通过以下的说明书而变得容易理解。
附图说明
[0015]附图示例性地示出了实施例并且构成说明书的一部分，与说明书的文字描述一起用于讲解实施例的示例性实施方式。所示出的实施例仅出于例示的目的，并不限制权利要求的范围。在所有附图中，相同的附图标记指代类似但不一定相同的要素。
[0016]图1示出了根据本公开的实施例的可以在其中实施本文描述的各种方法的示例性系统的示意图；
[0017]图2示出了根据本公开的实施例的用于自动驾驶的场景编码模型训练方法的流程图；
[0018]图3示出了根据本公开的实施例的用于自动驾驶的场景编码方法的流程图；
[0019]图4示出了根据本公开的实施例的用于自动驾驶的场景编码模型训练装置的结构框图；
[0020]图5示出了根据本公开的实施例的用于自动驾驶的场景编码装置的结本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种用于自动驾驶的场景编码模型训练方法，所述方法包括：获取样本数据集，所述样本数据集包括对应于多个第一时刻的多个第一样本数据，每个第一样本数据包括该第一样本数据相应的第一时刻对应的第一数据以及相应第一时刻之前的至少一个历史时刻对应的至少一个第二数据，所述第一数据包括相应第一时刻的车辆行驶状态数据，所述至少一个第二数据中的每个第二数据包括相应历史时刻的车辆行驶状态数据；基于至少一个掩码策略，对所述多个第一样本数据进行处理，以获取多个第二样本数据，所述至少一个掩码策略包括第一掩码策略，所述第一掩码策略用于对所述多个第一样本数据中的至少一个第三样本数据中每个第三样本数据的第一数据进行掩码处理；将所述多个第二样本数据输入所述场景编码模型，以获取所述场景编码模型输出的分别对应于所述多个第二样本数据的多个行驶场景编码；以及基于所述多个行驶场景编码训练所述场景编码模型。2.根据权利要求1所述的方法，其中，所述至少一个掩码策略从预设掩码策略集中获得，所述预设掩码策略集还包括第二掩码策略，所述第二掩码策略用于对所述多个第一样本数据中的至少一个第四样本数据中每个第四样本数据的至少一个第二数据进行随机掩码处理。3.根据权利要求2所述的方法，其中，所述至少一个第二数据数量为多个，所述预设掩码策略集还包括第三掩码策略，所述第三掩码策略用于对所述多个第一样本数据中的至少一个第五样本数据中每个第五样本数据的第三数据进行掩码处理，其中，所述第三数据为该第五样本数据对应的至少一个历史时刻中的最早历史时刻对应的第二数据。4.根据权利要求3所述的方法，还包括：将所述多个第一样本数据划分为多个样本批次；并且所述基于至少一个掩码策略，对所述多个第一样本数据进行处理，以获取多个第二样本数据包括：针对所述多个样本批次中的每个样本批次，基于预设概率在所述预设掩码策略集中随机选择至少一个掩码策略，对该样本批次中的样本数据进行处理，以获取经过处理的多个样本批次的样本数据，作为所述多个第二样本数据。5.根据权利要求1至4中任一项所述的方法，还包括：将所述多个第一样本数据划分为至少一个样本对，所述至少一个样本对中每个样本对中的两个样本数据对应的车辆行驶场景的相似度符合大于预设值；并且，所述基于所述多个行驶场景编码训练所述场景编码模型包括：基于所述多个行驶场景编码中的至少一个场景编码对中的两个场景编码之间的距离，训练所述场景编码模型，其中，所述至少一个场景编码对与所述至少一个样本对对应。6.根据权利要求1至5中任一项所述的方法，其中，所述第一数据包括相应第一时刻对应的障碍物状态数据以及路网数据，所述至少一个第二数据中的每个第二数据包括相应历史时刻的障碍物状态数据以及路网数据。7.一种用于自动驾驶的场景编码方法，所述方法包括：获取当前时刻的第一数据以及所述当前时刻之前的至少一个历史时刻对应的至少一个第二数据，所述第一数据包括所述当前时刻的车辆行驶状态数据、障碍物状态数据和路
网数据中的至少一者，所述至少一个第二数据中的每个第二数据包括相应历史时刻的车辆行驶状态数据、障碍物状态数据和路网数据中的至少一者；将所述第一数据和所述至少一个第二数据输入场景编码模型中，以获得所述场景编码模型输出的所述当前时刻的行驶场景编码，其中，所述场景编码模型根据权利要求1
‑
6中任一项所述的方法训练获得。8.一种用于自动驾驶的场景编码模型训练装置，所述装置包括：第一获取单元，被配置为获取样本数据集，所述样本数据集包括对应于多个第一时刻的多个第一样本数据，每个第一样本数据包括该第一样本数据相应的第一时刻对应的第一数据以及相应第一时刻之前的至少一个历史时刻对应的至少一个第二数据，所述第一数据包括相应第一时刻的车辆行驶状态数据，所述至少一个第...

【专利技术属性】
技术研发人员：宋泽良，柳长春，苏亚博，
申请(专利权)人：北京百度网讯科技有限公司，
类型：发明
国别省市：