关键词:
机器视觉
低照度
目标检测
轻量级算法
YOLOv7
摘要:
低照度目标检测是目标检测任务中常见的挑战之一。通用的目标检测方法在低照度条件下性能会明显下降,而现有的低照度目标检测方法会造成大量的计算资源消耗,并不适合部署在计算能力受限的设备上。为应对上述问题,提出一种端到端的轻量级目标检测算法LL-YOLO。针对低照度图像中特征信息不明显、难以学习与辨识的问题,设计低照度图像生成算法,通过生成低照度图像来训练检测器,帮助其学习低照度环境下的特征信息;并对检测器网络结构进行调整,减少特征信息在计算过程中的损失,提高模型对特征信息的敏感度。针对低照度图像中特征信息受噪声影响严重的问题,提出聚合周边信息的A-ELAN模块,使用深度可分离卷积与注意力机制捕获周边信息,增强获得的特征信息,减弱噪声的影响。实验结果表明,LL-YOLO算法在低照度目标检测数据集ExDark上平均精度均值(mAP@0.5)达到81.1%,相较直接训练的YOLOv7-tiny算法提高11.9百分点,相比于其他算法具有较强竞争力。