支持抗光照目标检测的改进YOLO算法 |
| |
引用本文: | 姚宇捷,彭育辉,陈泽辉,何维堃,吴庆,黄炜,陈文强.支持抗光照目标检测的改进YOLO算法[J].汽车工程,2023(5):777-785. |
| |
作者姓名: | 姚宇捷 彭育辉 陈泽辉 何维堃 吴庆 黄炜 陈文强 |
| |
作者单位: | 1. 福州大学机械工程及自动化学院;2. 福建汉特云智能科技有限公司 |
| |
基金项目: | 福建省自然科学基金(2021J01559)资助; |
| |
摘 要: | 针对现有的深度学习目标检测算法中存在的复杂光照场景下检测精度不高、实时性差等问题,提出了一种基于YOLO算法的抗光照目标检测网络模型YOLO-RLG。首先,将输入模型的RGB数据转换为HSV数据,从HSV数据分离出抗光照能力强的S通道,并与RGB数据合并生成RGBS数据,使输入数据具备抗光照能力;其次,将YOLOV4的主干网络替换成Ghostnet网络,并对其在普通卷积与廉价卷积的模型分配比例上进行调整,在保证检测精度的同时提高检测速度;最后,用EIoU替换CIoU改进模型的损失函数,提高了目标检测精度和算法鲁棒性。基于KITTI与VOC数据集的实验结果表明,与原网络模型比较,FPS提高了22.54与17.84 f/s,模型降低了210.3 M,精确度(AP)提升了0.83%与1.31%,且算法的抗光照能力得到显著增强。
|
关 键 词: | 机器视觉 抗光照图像处理 Ghostnet网络 损失函数 |
|
|