一种数据平衡和多任务的信号灯检测方法

本发明提供了一种数据平衡和多任务的信号灯检测方法，包括数据平衡、模型构建和训练测试。数据平衡通过采集源数据集和背景数据集，增强源数据和仿真数据。模型构建涉及特征编码器、颜色、方向和位置检测头。训练测试包括迁移学习、联合损失优化和单模块及联合测试。本发明可以提高信号灯检测的准确性和鲁棒性，提升交通监管系统的智能化水平。