基于YOLO-SCAT的可见光图像无人机小目标检测算法

A Small Object of UAV Detection Algorithm for Visible Light Images Based on YOLO-SCAT

  • 摘要: 非法使用民用低空小型无人机事件频发,严重威胁国家和社会和谐发展。针对基于可见光的无人机小目标检测存在漏检、误检和检测平均精度不高等问题,提出了YOLO-SCAT(YOLO-SCConv ATT)算法模型,通过重构ELAN(efficient layer aggregation networks)结构,减少空间和通道冗余特征;同时引入注意力机制CBAM(convolutional block attention module),训练中在空间和通道维度加强特征提取,提高模型检测平均精度。实验表明YOLO-SCAT模型的精确率、召回率、F1分数、mAP@0.5和mAP@0.5:0.95分别可达94.4%、94.4%、94.4%、94.7%和52.9%,证实了YOLO-SCAT模型提高了在复杂可见光场景下对无人机小目标的检测识别能力,能更好地满足反无人机系统的实际需求。

     

    Abstract: The frequent illegal use of civilian low altitude small unmanned aerial vehicles poses a serious threat to the harmonious development of the country and society. In response to the problems of missed detection, false detection, and low detection accuracy in visible light based small target detection of unmanned aerial vehicles, this paper proposes the YOLO SCAT (YOLO SCConv ATT) algorithm model to reconstruct the ELAN structure and reduce redundant spatial and channel features; At the same time, the attention mechanism CBAM is introduced to enhance feature extraction in space and channels during training, improving the detection accuracy of the model. The experiment shows that the YOLO-SCAT model has high accuracy, recall, F1 score mAP@0.5 The YOLO-SCAT model achieves 94.4%, 94.4%, 94.4%, 94.7%, and 52.9% for mAP @ 0.5:0.95, indicating that it improves the detection and recognition ability of small targets in complex visible light scenes, and can better meet the practical needs of anti drone systems.

     

/

返回文章
返回