基于机器学习模型的驾驶场景拓扑推理方法及电子设备

本申请公开了一种基于机器学习模型的驾驶场景拓扑推理方法及电子设备，模型包括以级联方式设置的多个循环模块，各循环模块包括交通元素检测模块和车道中心线检测模块，车道中心线检测模块包括车道中心线解码模块和车道中心线拓扑推理模块。该方法中，各级车道中心线解码模块基于输入的车道中心线特征信息和车道中心线拓扑推理信息，得到输入至下一级车道中心线解码模块和同级车道中心线拓扑推理模块的更新的车道中心线特征信息，以及注意力权重信息，如此实现循环交互，直至得到目标驾驶场景对应的拓扑关系。提升了驾驶场景拓扑关系推理的准确性或者精度，从而提升了自动驾驶控制方面的准确性。