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

Bird Detection Algorithm for Small Infrared Targets at an Airport 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 the accurate real-time monitoring of birds around airports is a critical issue in aviation safety that needs to be resolved. To address problems such as the low location accuracy of small bird targets in infrared monitoring at airports owing to background interference and low signal-to-noise ratios, this study developed a lightweight detection model (BISTD-YOLO) based on YOLOv8n. First, to reduce the model parameters and enhance its ability to capture small target details, a lightweight multiscale convolutional module (C2f-LMSC) and an edge-guided module (C2f-E-LMSC) were designed for the backbone and neck structures, respectively. Second, the network neck and detection head structures were reconstructed to retain shallow detailed information while reducing model complexity. Subsequently, a cross-space attention-enhanced feature pyramid structure (CSAEFPN) was built. It effectively suppressed background interference and solved the feature loss problem through the synergy between the BiFPN and SimAM. Finally, the normalized Wasserstein distance loss function was introduced to replace the traditional IoU metric and optimize the location accuracy for small targets. Experiments were performed using a self-constructed AIBD. The results showed that compared to YOLOv8n, the proposed model increased the mAP@0.5 and mAP@0.5:0.95 results by 5.5% and 2.2%, respectively, decreased the number of parameters by 38.2%, and was only 4 MB in size. In generalization tests, it boosted the mAP@0.5 values by 2.5%, 1.2%, and 0.6% on NUAA-SIRST, NUDT-SIRST, and IRSTD-1k, respectively.

     

/

返回文章
返回