信号灯控制模型训练方法、装置、电子设备及存储介质

本公开提供了信号灯控制模型训练方法、装置、电子设备及存储介质，涉及人工智能技术领域，尤其涉及深度学习、大模型、计算机视觉等技术领域，可应用于智能交通等场景。具体实现方案为：在预设的模拟路口环境中，获取视觉图像、俯瞰图像和交通信息；基于视觉图像、交通信息以及预设的实体类别，构建多种子任务训练样本；基于多种子任务训练样本，对初始模型中的投影层和映射层进行第一阶段训练，生成中间模型；基于视觉图像和俯瞰图像，构建决策推理训练样本；基于决策推理训练样本，对中间模型中的投影层和映射层进行第二阶段训练，生成信号灯控制模型。