关键词:
YOLOv8
行人检测
注意力机制
低照度
摘要:
在井下电机车无人驾驶障碍物检测任务中,由于光照不均、粉尘等因素的影响,电机车障碍物图像目标检测系统会出现提取目标特征困难、目标识别精度降低等问题,易导致障碍物误检和漏检。针对上述问题,提出一种基于SCI-YOLOv8的低照度目标检测算法,将SCINet自校正照明网络与YOLOv8目标检测算法相融合,使YOLOv8算法更有利于低光照目标检测。并将LSKA注意力机制嵌入到YOLOv8网络中Head部分C2f的末端,降低计算和内存成本的同时,保持了高效的图像处理能力。实验结果表明,本研究所提出的算法在公共低光数据集Exdark目标检测中,mAP@50为57.7%,mAP@50∶95为35.4%。相较于原始YOLOv8目标检测算法mAP@50提高了1个百分点,mAP@50∶95提高了1.4个百分点。在井下低光数据集LLP目标检测中,SCI-YOLOv8模型的mAP@50达到97.3%,mAP@50∶95为68.2%,相较于原始模型分别提高了3.4个百分点和8.6个百分点。本研究所提出的SCI-YOLOv8算法在低光场景的目标检测中具有优越性,能满足井下低光场景的目标检测任务要求,为井下电机车安全、高效、智能运行提供了技术支撑。