基于BISTD-YOLO的机场鸟类红外小目标检测算法

Bird Detection Algorithm for Airport Infrared Small Targets Based on BISTD-YOLO

  • 摘要: 鸟类活动对航空安全构成严重威胁,实现机场周边鸟类的实时精准监测是航空安全领域亟待解决的关键问题。针对当前机场红外监测场景中背景干扰和低信噪比导致的鸟类小目标定位精度不足等问题,本文基于YOLOv8n提出了一种轻量化的检测模型BISTD-YOLO。首先,为了降低模型参数并增强小目标细节的捕获能力,分别在Backbone和Neck结构中设计了轻量级多尺度卷积模块C2f-LMSC和边缘引导模块C2f-E-LMSC;其次,通过重构网络颈部和检测头结构,在保留浅层细节信息的同时降低模型复杂度;然后,构建了跨空间注意力增强特征金字塔CSAEFPN结构,通过BiFPN与SimAM协同优化,有效抑制复杂背景干扰并解决特征丢失问题;最后,引入归一化Wasserstein距离损失函数替代传统IoU指标,优化小目标定位精度。实验在自建的机场红外鸟类数据集AIBD上进行,结果表明,与YOLOv8n相比,所提模型的mAP@0.5和mAP@0.5:0.95分别提升了5.5%和2.2%,参数量减少了38.2%,模型大小仅为4MB。在泛化测试中,所提模型在NUAA-SIRST、NUDT-SIRST和IRSTD-1k数据集上的mAP@0.5分别提升了2.5%、1.2%和0.6%。

     

    Abstract: Bird activity poses a severe threat to aviation safety, and real-time, accurate monitoring of birds around airports is a critical issue in aviation safety that needs to be resolved. To address problems such as low location accuracy of small bird targets in infrared monitoring at airports due to background interference and low signal-to-noise ratios, this paper proposes a lightweight detection model, BISTD-YOLO, based on YOLOv8n. Firstly, to reduce model parameters and enhance the ability to capture small target details, a lightweight multi-scale convolutional module C2f-LMSC and an edge-guided module C2f-E-LMSC are designed in the Backbone and Neck structures respectively. Secondly, the network neck and detection head structures are reconstructed to retain shallow detail information while reducing model complexity. Then, a cross-space attention enhanced feature pyramid structure CSAEFPN is built. It effectively suppresses background interference and solves the feature loss problem through the synergy of BiFPN and SimAM. Finally, the normalized Wasserstein distance loss function is introduced to replace the traditional IoU metric to optimize the location accuracy of small targets. Experiments are carried out on the self-constructed AIBD. Results show that, compared to YOLOv8n, the proposed model increases mAP@0.5 and mAP@0.5:0.95 by 5.5% and 2.2%, reduces parameters by 38.2%, and is only 4MB in size. In generalization tests, it boosts mAP@0.5 by 2.5%, 1.2%, and 0.6% on NUAA-SIRST, NUDT-SIRST, and ISTD-1k respectively.

     

/

返回文章
返回