摘要

针对视频图像中战斗机起落架目标小、清晰度低、检测难度大等问题,提出了改进的YOLOv5目标检测算法来完成对战斗机起落架的检测。为提高起落架检测的准确性,将自动色彩均衡算法应用于视频图像中提高视频图像的清晰度,在YOLOv5的主干特征提取网络提取的三个有效特征层后嵌入高效通道注意力机制使网络将注意力集中到图像目标上,忽略背景信息的干扰。实验结果表明,改进后的YOLOv5模型与原YOLOv5模型相比,召回率提高了0.52%,精确率提高了1.52%,mAP提高了1.43%,在满足实时性的同时提高了对战斗机起落架的检测精度。