改进YOLOv5的机场跑道异物目标检测算法

doi:10.3778/j.issn.1002-8331.2207-0462

摘要/Abstract

摘要： 针对机场跑道异物（foreign object debris，FOD）在图像中目标占比小，特征不明显，经常导致误检、漏检的问题，提出一种改进YOLOv5的FOD目标检测算法。改进多尺度融合与检测部分，融合高分辨率特征图增强小目标特征表达，移除大目标检测层以减少网络推理计算量；引入轻量高效的卷积注意力模块（CBAM），从空间与通道两个维度提升模型关注目标特征的能力；在特征融合阶段采用RepVGG模块，提高模型特征融合能力的同时提高了检测精度；采用SIoU Loss作为损失函数，提升了边框回归的速度与精度。在自制FOD数据集上进行对比实验，结果表明：该方法在满足实时性的条件下，实现了95.01%的mAP50、55.79%的mAP50：95，比原算法YOLOv5分别提高了2.78、3.28个百分点，有效解决了传统FOD检测误检、漏检问题，同时与主流目标检测算法相比，提出的改进算法更适用于FOD检测任务。

关键词: 机场跑道异物, YOLOv5, CBAM注意力模块, RepVGG模块

Abstract: Aiming at the problem that the foreign object debris（FOD） of the airport runway has a small proportion of the target in the image and the features are not obvious, which often leads to false detection and missed detection, an improved YOLOv5 FOD target detection algorithm is proposed. Firstly, it improves the multi-scale fusion and detection part, fuses high-resolution feature maps to enhance the feature expression of small targets, and removes the large target detection layer to reduce the computational complexity of network reasoning. Secondly, a lightweight and efficient convolutional attention module（CBAM） is introduced. It improves the ability of the model to focus on target features from the two dimensions of space and channel. Then it uses the RepVGG module in the feature fusion stage to improve the feature fusion ability of the model and improve the detection accuracy. Finally, it uses SIoU Loss as the loss function to improve the speed and precision of bounding box regression. Comparative experiments are carried out on the self-made FOD data set. The results show that the method achieves 95.01% mAP50 and 55.79% mAP50∶95 under the condition of real-time performance, which is 2.78 and 3.28 percentage points higher than the original algorithm YOLOv5, respectively. It effectively solves the problems of false detection and missed detection of traditional FOD detection. At the same time, compared with mainstream target detection algorithms, the proposed improved algorithm is more suitable for FOD detection tasks.

Key words: foreign object debris（FOD）, YOLOv5, CBAM attention module, RepVGG module

李小军, 邓月明, 陈正浩, 何鑫. 改进YOLOv5的机场跑道异物目标检测算法[J]. 计算机工程与应用, 2023, 59(2): 202-211.

LI Xiaojun, DENG Yueming, CHEN Zhenghao, HE Xin. Improved YOLOv5’s Foreign Object Debris Detection Algorithm for Airport Runways[J]. Computer Engineering and Applications, 2023, 59(2): 202-211.

参考文献

[1] 国家民航总局机场司民航安全技术中心.FOD防范手册[Z].北京：国家民航总局机场司民航安全技术中心，2009：1-5.
Civil Aviation Safety Technology Center，Airport Division，General Administration of Civil Aviation of China.FOD prevention manual[Z].Beijing：Civil Aviation Safety Technology Center，Airport Division，General Administration of Civil Aviation of China，2009：1-5.
[2] 李华琼，张中仅，王雨果，等.CFAR方法在机场跑道FOD检测中的性能分析[J].无线电工程，2015，45（9）：53-57.
LI H Q，ZHANG Z J，WANG Y G，et al.Performance analysis and comparison of CFAR methods for FOD detection in airport runway environment[M].Radio Engineering，2015，45（9）：53-57.
[3] WU J，WANG H，YU X，et al.CFAR detection method in multi-target environments for foreign object debris surveillance radar[C]//Proceedings of the Second International Conference on Communications，Signal Processing，and Systems.Cham：Springer，2014：533-540.
[4] WAN Y，LIANG X，BU X，et al.FOD detection method based on iterative adaptive approach for millimeter-wave radar[J].Sensors，2021，21（4）：1241.
[5] QIN F，BU X，LIU Y，et al.Foreign object debris automatic target detection for millimeter-wave surveillance radar[J].Sensors，2021，21：3853.
[6] WANG Y，SONG Q，WANG J，et al.A correction method to systematic phase drift of a high resolution radar for foreign object debris detection[J].Remote Sensing，2022，14（8）：1787.
[7] CAO X，WANG P，MENG C，et al.Region based CNN for foreign object debris detection on airfield pavement[J].Sensors，2018，18（3）：737.
[8] 蒋进，唐碧蔚，宋少雷，等.基于Faster-RCNN的机场跑道异物识别定位技术研究[J].航空电子技术，2020，51（3）：8-12.
JIANG J，TANG B W，SONG S L，et al.Research on identification and location of foreign object debris on airport runway based on Faster-RCNN[J].Avionics Technology，2020，51（3）：8-12.
[9] GIRSHICK R.Fast R-CNN[C]//Proceedings of the IEEE International Conference on Computer Vision，2015：1440-1448.
[10] REN S，HE K，GIRSHICK R，et al.Faster R-CNN：towards real-time object detection with region proposal networks[C]//Advances in Neural Information Processing Systems，2015.
[11] HE K，GKIOXARI G，DOLLáR P，et al.Mask R-CNN[C]//Proceedings of the IEEE International Conference on Computer Vision，2017：2961-2969.
[12] 郭晓静，隋昊达.改进YOLOv3在机场跑道异物目标检测中的应用[J].计算机工程与应用，2021，57（8）：249-255.
GUO X J，SUI H D.Application of improved YOLOv3 in foreign object debris target detection on airfield pavement[J].Computer Engineering and Applications，2021，57（8）：249-255.
[13] LI P，LI H.Research on fod detection for airport runway based on yolov3[C]//2020 39th Chinese Control Conference（CCC），2020：7096-7099.
[14] 岳晓新，贾君霞，陈喜东，等.改进YOLO V3的道路小目标检测[J].计算机工程与应用，2020，56（21）：218-223.
YUE X X，JIA J X，CHEN X D，et al.Road small target detection algorithm based on improved YOLO V3[J].Computer Engineering and Applications，2020，56（21）：218-223.
[15] REDMON J，FARHADI A.Yolov3：an incremental improvement[J].arXiv：1804.02767，2018.
[16] BOCHKOVSKIY A，WANG C Y，LIAO H Y M.Yolov4：optimal speed and accuracy of object detection[J].arXiv：2004.10934，2020.
[17] WOO S，PARK J，LEE J Y，et al.CBAM：convolutional block attention module[C]//Proceedings of the European Conference on Computer Vision（ECCV），2018：3-19.
[18] DING X，ZHANG X，MA N，et al.Repvgg：making vgg-style convnets great again[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition，2021：13733-13742.
[19] GEVORGYAN Z.SIoU Loss：more powerful learning for bounding box regression[J].arXiv：2205.12740，2022.
[20] HU J，SHEN L，SUN G.Squeeze-and-excitation networks[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition，2018：7132-7141.
[21] WANG Q，WU B，ZHU P，et al.Supplementary material for “ECA-Net：efficient channel attention for deep convolutional neural networks”[C]//Proceedings of the 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition，Seattle，WA，USA，2020：13-19.
[22] HOU Q，ZHOU D，FENG J.Coordinate attention for efficient mobile network design[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition，2021：13713-13722.
[23] LIN T Y，GOYAL P，GIRSHICK R，et al.Focal loss for dense object detection[C]//Proceedings of the IEEE International Conference on Computer Vision，2017：2980-2988.
[24] CAI Z，VASCONCELOS N.Cascade R-CNN：delving into high quality object detection[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition，2018：6154-6162.
[25] TIAN Z，SHEN C，CHEN H，et al.FCOS：a simple and strong anchor-free object detector[J].IEEE Transactions on Pattern Analysis and Machine Intelligence，2022，44（4）：1922-1933.

[26] GE Z，LIU S，WANG F，et al.Yolox：exceeding yolo series in 2021[J].arXiv：2107.08430，2021.

[27] WANG C Y，YEH I H，LIAO H Y M.You only learn one representation：unified network for multiple tasks[J].arXiv：2105.04206，2021.