基于梯形跨尺度特征耦合网络的SAR图像舰船检测

doi:10.19678/j.issn.1000-3428.0063327

摘要/Abstract

摘要： 在合成孔径雷达（SAR）图像舰船检测中，现有检测方法难以有效提取多尺度语义信息，无法准确地表示其在整个网络中的信息权重，且定位模块与分类模块相关性较弱，导致定位不准确。提出一种梯形跨尺度特征耦合网络，通过梯形特征金字塔网络提取各级语义信息，采用交叉结构代替跳连结构，提高网络的泛化能力和语意表征能力，并引入可训练权重因子表示各级语义信息的重要性。在此基础上，将定位模块与分类模块通过耦合检测头增强两者之间的相关性，引入可变形卷积对最终的定位输出进行二次校准，从而提高检测精度。实验结果表明，与FasterRCNN、CascadeRCNN、RetinaNet等主流网络相比，该网络在SSDD数据集上的检测精度提高了2.74个百分点以上，具有良好的检测性能。在近岸复杂场景下，该网络能更有效地检测密集目标和多尺度目标，降低误检和漏检的概率。

关键词: 舰船检测, 梯形特征金字塔, 多尺度特征聚合, 耦合网络, 可训练权重因子

Abstract: In ship detection in Synthetic Aperture Radar(SAR) images, effectively extracting multi-scale semantic information by using existing detection methods is difficult.Additionally, the information weight in the whole network cannot be accurately represented, and the correlation between the positioning module and the classification module is weak, leading to inaccurate positioning.This study presents a trapezoidal cross-scale feature-coupling network, in which the semantic information at all levels is extracted by employing a trapezoidal feature pyramid network.The cross structure is used to replace the jump structure to improve the generalization ability and semantic representation ability of the network.A trainable weight factor is introduced to represent the importance of semantic information at all levels.The correlation between the positioning module and the classification module can be enhanced by coupling the detection head, and a deformable convolution is introduced to calibrate the final positioning output and improve the detection accuracy.Experimental results show that compared with mainstream networks, such as the FasterRCNN, CascadeRCNN, and RetinaNet, the detection accuracy of the proposed network on the SSDD dataset is improved by more than 2.74 percentage points, demonstrating good detection performance.In the near-shore complex scene, the network can detect dense and multi-scale targets more effectively and reduce the probability of false or missed detections.

Key words: ship detection, trapezoidal characteristic pyramid, multiscale characteristic polymerization, coupling network, trainable weight factor

中图分类号:

TP393

黄帅, 张毅. 基于梯形跨尺度特征耦合网络的SAR图像舰船检测[J]. 计算机工程, 2022, 48(12): 270-280.

HUANG Shuai, ZHANG Yi. Ship Detection in SAR Images Based on Trapezoidal Cross-scale Feature-coupling Network[J]. Computer Engineering, 2022, 48(12): 270-280.

http://www.ecice06.com/CN/Y2022/V48/I12/270

图/表 15

20230112184525

20230112184530

20230112184533

20230112184537

20230112184540

20230112184544

20230112184547

20230112184550

20230112184554

20230112184557

20230112184601

20230112184605

20230112184609

20230112184612

20230112184616

参考文献

[1] HATEM G M, ABDUL SADAH J W, SALMAN J, et al.Improve CFAR algorithm based on closed loop by neural network[C]//Proceedings of the 4th Scientific International Conference Najaf.Washington D.C., USA:IEEE Press, 2019:15-19.
[2] CHEN J X, ZHOU S H, VARSHNEY P K, et al.A sparsity based CFAR algorithm for dense radar targets[C]//Proceedings of IEEE Radar Conference.Washington D.C., USA:IEEE Press, 2020:1-6.
[3] AI J Q, PEI Z L, YAO B D, et al.AIS data aided Rayleigh CFAR ship detection algorithm of multiple-target environment in SAR images[C]//Proceedings of IEEE Transactions on Aerospace and Electronic Systems.Washington D.C., USA:IEEE Press, 2021:1266-1282.
[4] AI J Q, MAO Y X, LUO Q W, et al.Robust CFAR ship detector based on bilateral-trimmed-statistics of complex ocean scenes in SAR imagery:a closed-form solution[C]//Proceedings of IEEE Transactions on Aerospace and Electronic Systems.Washington D.C., USA:IEEE Press, 2021:1872-1890.
[5] LIAO M S, WANG C C, WANG Y, et al.Using SAR images to detect ships from sea clutter[C]//Proceedings of IEEE Geoscience and Remote Sensing Letters.Washington D.C., USA:IEEE Press, 2008:194-198.
[6] LI H C, HONG W, WU Y R, et al.On the empirical-statistical modeling of SAR images with generalized gamma distribution[J].IEEE Journal of Selected Topics in Signal Processing, 2011, 5(3):386-397.
[7] GAO G, OUYANG K W, LUO Y B, et al.Scheme of parameter estimation for generalized gamma distribution and its application to ship detection in SAR images[C]//Proceedings of IEEE Transactions on Geoscience and Remote Sensing.Berlin, Germany:Springer, 2016:1812-1832.
[8] LAW H, DENG J.Cornernet:detecting objects as paired keypoints[C]//Proceedings of the European Conference on Computer Vision.Washington D.C., USA:IEEE Press, 2018, 734-750.
[9] ZHOU X Y, ZHUO J C, KRÄHENBÜHL P.Bottom-up object detection by grouping extreme and center points[C]//Proceedings of IEEE/CVF Conference on Computer Vision and Pattern Recognition.Washington D.C., USA:IEEE Press, 2019:850-859.
[10] DUAN K W, BAI S, XIE L X, et al.CenterNet:keypoint triplets for object detection[C]//Proceedings of IEEE/CVF International Conference on Computer Vision.Washington D.C., USA:IEEE Press, 2019:6568-6577.
[11] FU J M, SUN X, WANG Z R, et al.An anchor-free method based on feature balancing and refinement network for multiscale ship detection in SAR images[J].IEEE Transactions on Geoscience and Remote Sensing, 2021, 59(2):1331-1344.
[12] GAO F, HE Y S, WANG J, et al.Anchor-free convolutional network with dense attention feature aggregation for ship detection in SAR images[J].Remote Sensing, 2020, 12(16), 19-24.
[13] MAO Y X, YANG Y Q, MA Z Y, et al.Efficient low-cost ship detection for SAR imagery based on simplified U-net[J].IEEE Access, 2012:8:69742-69753.
[14] CUI Z Y, WANG X Y, LIU N Y, et al.Ship detection in large-scale SAR images via spatial shuffle-group enhance attention[J].IEEE Transactions on Geoscience and Remote Sensing, 2021, 59(1):379-391.
[15] WANG J, CHEN J, WANG P B, et al.An algorithm for azimuth ambiguities detection in SAR images using faster-RCNN[C]//Proceedings of the 6th Asia-Pacific Conference on Synthetic Aperture Radar.Washington D.C., USA:IEEE Press, 2019:1-5.
[16] GUI Y C, LI X H, XUE L, et al.A scale transfer convolution network for small ship detection in SAR images[C]//Proceedings of the 8th Joint International Information Technology and Artificial Intelligence Conference.Washington D.C., USA:IEEE Press, 2019:1845-1849.
[17] SERMANET P, EIGEN D, ZHANG X, et al.OverFeat:integrated recognition, localization and detection using convolutional networks[EB/OL].[2021-10-20].https://arxiv.org/abs/1312.6229.
[18] ZHOU L, WEI S Y, CUI Z M, et al.Lira-YOLO:a lightweight model for ship detection in radar images[J].Journal of Systems Engineering and Electronics, 2020, 31(5):950-956.
[19] FU C Y, LIU W, RANGA A, et al.DSSD:deconvolutional single shot detector[EB/OL].[2021-10-20].https://arxiv.org/abs/1701.06659.
[20] LIU W, ANGUELOV, ERHAN D, et al.SSD:single shot multibox detector[C]//Proceedings of European Conference on Computer Vision.Berlin, Germany:Springer, 2016:21-37.
[21] TAN M X, PANG R M, LE Q V.EfficientDet:scalable and efficient object detection[C]//Proceedings of IEEE/CVF Conference on Computer Vision and Pattern Recognition.Washington D.C., USA:IEEE Press, 2016:10778-10787.
[22] WANG T, ZHU X G, PANG J M, et al.FCOS3D:fully convolutional one-stage monocular 3D object detection[C]//Proceedings of IEEE/CVF International Conference on Computer Vision Workshops.Washington D.C., USA:IEEE Press, 2019:913-922.
[23] REDMON J, FARHADI A.YOLO9000:better, faster, stronger[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition.Washington D.C., USA:IEEE Press, 2017:6517-652.
[24] LONG X, DENG K P, WANG G Z, et al.PP-YOLO:an effective and efficient implementation of object detector[EB/OL].[2021-10-20].https://arxiv.org/abs/2007.12099.
[25] REDMON J, DIVVALA S, GIRSHICK R, et al.You only look once:unified, real-time object detection[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition.Washington D.C., USA:IEEE Press, 2016:779-788.
[26] LIN T Y, GOYAL P, GIRSHICK R, et al.Focal loss for dense object detection[C]//Proceedings of IEEE International Conference on Computer Vision.Washington D.C., USA:IEEE Press, 2017:2999-3007.
[27] SHRIVASTAVA A, GUPTA A, GIRSHICK R.Training region-based object detectors with online hard example mining[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition.Washington D.C., USA:IEEE Press, 2016:761-769.
[28] HE Y H, ZHU C C, WANG J R, et al.Bounding box regression with uncertainty for accurate object detection[C]//Proceedings of IEEE/CVF Conference on Computer Vision and Pattern Recognition.Washington D.C., USA:IEEE Press, 2019:2883-2892.
[29] WU S, LI X, WANG X.IoU-aware single-stage object detector for accurate localization[J].Image and Vision Computing, 2020, 97(5), 11-20.
[30] JIANG B, LUO R., MAO J, et al.Acquisition of localization confidence for accurate object detection[C]//Proceedings of European Conference on Computer Vision.Berlin, Germany:Springer, 2018:784-799.
[31] BODLA N, SINGH B, CHELLAPPA R, et al.Soft-NMS-improving object detection with one line of code[C]//Proceedings of IEEE International Conference on Computer Vision.Washington D.C., USA:IEEE Press, 2017:5562-5570.
[32] YU J H, JIANG Y N, WANG Z Y, et al.UnitBox:an advanced object detection network[C]//Proceedings of the 24th ACM International Conference on Multimedia.New York, USA:ACM Press, 2016:516-520.
[33] TYCHSEN-SMITH L, PETERSSON L.Improving object localization with fitness NMS and bounded IoU loss[C]//Proceedings of IEEE/CVF Conference on Computer Vision and Pattern Recognition.Washington D.C., USA:IEEE Press, 2018:6877-6885.
[34] LIN T Y, DOLLÁR P, GIRSHICK R, et al.Feature pyramid networks for object detection[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition.Washington D.C., USA:IEEE Press, 2017:936-944.
[35] GHIASI G, LIN T Y, LE Q V.NAS-FPN:learning scalable feature pyramid architecture for object detection[C]//Proceedings of IEEE/CVF Conference on Computer Vision and Pattern Recognition.Washington D.C., USA:IEEE Press, 2019:7029-7038.
[36] LIU S, QI L, QIN H F, et al.Path aggregation network for instance segmentation[C]//Proceedings of IEEE/CVF Conference on Computer Vision and Pattern Recognition.Washington D.C., USA:IEEE Press, 2018:8759-8768.
[37] KIM S W, KOOK H K, SUN J Y, et al.Parallel feature pyramid network for object detection[C]//Proceedings of European Conference on Computer Vision.Berlin, Germany:Springer, 2018:234-250.
[38] WOO S, HWANG S, KWEON I.S.Stairnet:top-down semantic aggregation for accurate one shot detection[C]///Proceedings of 2018 IEEE Winter Conference on Applications of Computer Vision.Washington D.C., USA:IEEE Press, 2018:1093-1102.
[39] ZHOU P, NI B B, GENG C, et al.Scale-transferrable object detection[C]//Proceedings of IEEE/CVF Conference on Computer Vision and Pattern Recognition.Washington D.C., USA:IEEE Press, 2018:528-537.
[40] KONG T, SUN F C, YAO A B, et al.RON:reverse connection with objectness prior networks for object detection[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition.Washington D.C., USA:IEEE Press, 2017:5244-5252.
[41] HU J, SHEN L, SUN G.Squeeze-and-excitation networks[C]//Proceedings of IEEE/CVF Conference on Computer Vision and Pattern Recognition.Washington D.C., USA:IEEE Press, 2018:7132-7141.
[42] BELLO I, ZOPH B, LE Q, et al.Attention augmented convolutional networks[C]//Proceedings of IEEE/CVF International Conference on Computer Vision.Washington D.C., USA:IEEE Press, 2019:3285-3294.
[43] WOO S, PARK J, LEE J Y, et al.CBAM:convolutional block attention module[EB/OL].[2021-10-20].https://arxiv.org/abs/1807.06521.
[44] WANG Q L, WU B G, ZHU P F, et al.ECA-net:efficient channel attention for deep convolutional neural networks[C]//Proceedings of IEEE/CVF Conference on Computer Vision and Pattern Recognition.Washington D.C., USA:IEEE Press, 2020:11531-11539.
[45] HE K M, ZHANG X Y, REN S Q, et al.Deep residual learning for image recognition[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition.Washington D.C., USA:IEEE Press, 2016:770-778.
[46] CHEN K, CAO Y H, LOY C C, et al.Feature pyramid grids[EB/OL].[2021-10-20].https://arxiv.org/abs/2004.03580.
[47] DAI J F, QI H Z, XIONG Y W, et al.Deformable convolutional networks[C]//Proceedings of IEEE International Conference on Computer Vision.Washington D.C., USA:IEEE Press, 2017:764-773.
[48] 徐志京, 黄海.基于多重连接特征金字塔的SAR图像舰船目标检测[J].激光与光电子学进展, 2021, 58(8):287-294. XU Z J, HUANG H.Ship detection in SAR image based on multiple connected features pyramid network[J].Laser&Optoelectronics Progress, 2021, 58(8):287-294.(in Chinese)
[49] 侯笑晗, 金国栋, 谭力宁.基于深度学习的SAR图像舰船目标检测综述[J].激光与光电子学进展, 2021, 58(4):53-64. HOU X H, JIN G D, TAN L N.Survey of ship detection in SAR images based on deep learning[J].Laser & Optoelectronics Progress, 2021, 58(4):53-64.(in Chinese)
[50] 周雪珂, 刘畅, 周滨.多尺度特征融合与特征通道关系校准的SAR图像船舶检测[J].雷达学报, 2021, 10(4):531-543. ZHOU X K, LIU C, ZHOU B.Ship detection in SAR images based on multiscale feature fusion and channel relation calibration of features[J].Journal of Radars, 2021, 10(4):531-543.(in Chinese)
[51] 孙忠镇, 戴牧宸, 雷禹, 等.基于级联网络的复杂大场景SAR图像舰船目标快速检测[J].信号处理, 2021, 37(6):941-951. SUN Z Z, DAI M C, LEI Y, et al.Fast detection of ship targets for complex large-scene SAR images based on a cascade network[J].Journal of Signal Processing, 2021, 37(6):941-951.(in Chinese)

选择文件类型/文献管理软件名称

选择包含的内容