陈海永, 刘登斌, 晏行伟. 基于IDOU-YOLO的红外图像无人机目标检测算法[J]. 应用光学.
引用本文: 陈海永, 刘登斌, 晏行伟. 基于IDOU-YOLO的红外图像无人机目标检测算法[J]. 应用光学.
CHEN Haiyong, LIU Dengbin, YAN Xingwei. Infrared image UAV target detection algorithm based on IDOU-YOLO[J]. Journal of Applied Optics.
Citation: CHEN Haiyong, LIU Dengbin, YAN Xingwei. Infrared image UAV target detection algorithm based on IDOU-YOLO[J]. Journal of Applied Optics.

基于IDOU-YOLO的红外图像无人机目标检测算法

Infrared image UAV target detection algorithm based on IDOU-YOLO

  • 摘要: 低空小型无人机(unmanned aerial vehicle,UAV)侵扰敏感区域事件频发,使国家和社会面临严重安全威胁。针对基于热成像的无人机目标检测存在漏检率高、检测精度不足的问题,提出了IDOU-YOLO (infrared detection of UAV-YOLO)算法模型,通过构建多尺度融合特征金字塔机制,充分挖掘特征空间信息,聚焦尺度的信息融合及丰富模型的信息表征能力,增强目标检测能力;同时引入了边界框损失函数SIoU,在训练过程中提高模型的检测精度,加快模型的收敛速度。实验结果表明IDOU-YOLO模型的精确率、召回率、F1分数、mAP@0.5和mAP@0.5:0.95分别达到99.2%、96.3%、97.7%、98.4%和70.2%,表明IDOU-YOLO算法模型在红外无人机目标检测任务中具有显著优势和应用潜力。

     

    Abstract: Small, low-altitude small unmanned aerial vehicles (UAVs) frequently invade sensitive areas, posing a serious threat to national and social security. There are problems such as high missed detection rate and insufficient detection accuracy for UAV target detection based on thermal imaging. This paper proposes the IDOU-YOLO (Infrared Detection Of UAV-YOLO) algorithm model. A multi-scale merged feature pyramid mechanism is constructed to fully explore the feature space information, focuses on scale information fusion and the rich information representation ability of the model, and enhances target detection ability. The bounding box loss function SIOU is introduced to improve the detection accuracy of the model and accelerates the convergence speed of the model in the training process. The experimental results show that the precision, recall, F1 score, mAP@0.5 and mAP@0.5 0.95 achieved 99.2%, 96.3%, 97.7%, 98.4%, and 70.2%, indicating that the IDOU-YOLO algorithm model has significant advantages and application potential in thermal imaging based UAV target detection.

     

/

返回文章
返回