The research on lightweight SAR ship detection method based on regression model and attention

被引:2
作者
Li Li-Yuan [1 ,3 ]
Li Xiao-Yan [2 ]
Hu Zhuo-Yue [1 ]
Su Xiao-Feng [1 ]
Chen Fan-Sheng [1 ,2 ]
机构
[1] Chinese Acad Sci, Shanghai Inst Tech Phys, Key Lab Intelligent Infrared Percept, Shanghai 200083, Peoples R China
[2] Univ Chinese Acad Sci, Hangzhou Inst Adv Study, Hangzhou 310024, Peoples R China
[3] Univ Chinese Acad Sci, Beijing 100049, Peoples R China
基金
中国国家自然科学基金;
关键词
ship detection; YOLO regression model; channel attention; lightweight;
D O I
10.11972/j.issn.1001-9014.2022.03.013
中图分类号
O43 [光学];
学科分类号
070207 ; 0803 ;
摘要
Synthetic aperture radar(SAR)has the advantages of all-sky and all-weather earth observation without cloud interference. Ship detection based on SAR images has been widely used in civil and military fields,including maritime search and rescue,port reconnaissance,territorial sea defense. However,different from large ships,the misdetection rate of small ships with fewer pixels and lower contrast is high. And it is difficult to balance speed and accuracy during on-orbit ship detection. To solve the above problems,an improved lightweight ships detection method(ImShips)based on YOLOv5s is proposed. Firstly,the standard convolution with small receptive field is adopted at the bottom of the baseline to obtain spatial information about small ships. And the dilated convolution with enlarged receptive field is added at the top of the baseline to preserve more semantic features,which is conducive to extract large targets feature. Then,a lightweight channel attention mechanism is applied to the backbone and neck of YOLOv5. And the weight is allocated to filter more important texture information. Finally,the depth-wise separable convolution is adopted to replace the standard convolution during down-sampling to reduce the number of parameters and improve the inference speed. Compared with YOLOv5s model,the experimental results show that ImShips achieve an increase in AP,while the FLOPs are reduced by 45. 61%,and the speed is increased by 8. 31% in SSDD and ISSID datasets. The speed and accuracy of ImShips model are improved effectively on sea surface object detection. The proposed method has great application potential for on-orbit ship detection.
引用
收藏
页码:618 / 625
页数:8
相关论文
共 23 条
  • [1] [Anonymous], 2016, P INT C LEARN REPR
  • [2] Bochkovskiy A., 2020, YOLOv4: Optimal Speed and Accuracy of Object Detection, DOI DOI 10.48550/ARXIV.2004.10934
  • [3] Detection of Small Ship Objects Using Anchor Boxes Cluster and Feature Pyramid Network Model for SAR Imagery
    Chen, Peng
    Li, Ying
    Zhou, Hui
    Liu, Bingxin
    Liu, Peng
    [J]. JOURNAL OF MARINE SCIENCE AND ENGINEERING, 2020, 8 (02)
  • [4] Fast R-CNN
    Girshick, Ross
    [J]. 2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, : 1440 - 1448
  • [5] Rich feature hierarchies for accurate object detection and semantic segmentation
    Girshick, Ross
    Donahue, Jeff
    Darrell, Trevor
    Malik, Jitendra
    [J]. 2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, : 580 - 587
  • [6] GhostNet: More Features from Cheap Operations
    Han, Kai
    Wang, Yunhe
    Tian, Qi
    Guo, Jianyuan
    Xu, Chunjing
    Xu, Chang
    [J]. 2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, : 1577 - 1586
  • [7] Searching for MobileNetV3
    Howard, Andrew
    Sandler, Mark
    Chu, Grace
    Chen, Liang-Chieh
    Chen, Bo
    Tan, Mingxing
    Wang, Weijun
    Zhu, Yukun
    Pang, Ruoming
    Vasudevan, Vijay
    Le, Quoc V.
    Adam, Hartwig
    [J]. 2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, : 1314 - 1324
  • [8] Hu J, 2018, PROC CVPR IEEE, P7132, DOI [10.1109/TPAMI.2019.2913372, 10.1109/CVPR.2018.00745]
  • [9] A Large-Scale Database and a CNN Model for Attention-Based Glaucoma Detection
    Li, Liu
    Xu, Mai
    Liu, Hanruo
    Li, Yang
    Wang, Xiaofei
    Jiang, Lai
    Wang, Zulin
    Fan, Xiang
    Wang, Ningli
    [J]. IEEE TRANSACTIONS ON MEDICAL IMAGING, 2020, 39 (02) : 413 - 424
  • [10] Selective Kernel Networks
    Li, Xiang
    Wang, Wenhai
    Hu, Xiaolin
    Yang, Jian
    [J]. 2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, : 510 - 519