[发明专利]一种轻量级且高效的单阶段夜间车辆检测算法

摘要

本发明提供了一种轻量级且高效的单阶段夜间车辆检测算法，属于目标检测领域。本发明基于现有的YOLOv3目标检测网络进行改进，首先构建一个目前较为稀缺的夜间车辆数据集，该数据集包括各种场景，多种天气，能使模型具有较强的泛化能力，然后将数据集进行训练，经新提出的几个模块，使得网络在深层包含更多的语义信息，并且相互融合，更好的学习夜间车辆数据的特征，经过网络生成多个带置信度和预测类别的候选框，利用非极大抑制过滤多余预测框，形成最终的夜间车辆定位和识别结果。本发明通过多个融合信息的模块提升检测质量。本发明利用真实的图像数据集进行评估本发明，并观察在最先进的基线上的改进。