YOLO-RSA: A Multiscale Ship Detection Algorithm Based on Optical Remote Sensing Image

被引:7
作者
Fang, Zhou [1 ,2 ]
Wang, Xiaoyong [1 ]
Zhang, Liang [2 ]
Jiang, Bo [1 ]
机构
[1] Natl Ocean Technol Ctr, Tianjin 300112, Peoples R China
[2] Tianjin Univ, Coll Ocean Sci & Technol, Tianjin 300072, Peoples R China
基金
国家重点研发计划;
关键词
deep learning; multiscale ship detection; remote-sensing image; small ship attention mechanism; complex scenarios; OBJECT DETECTION;
D O I
10.3390/jmse12040603
中图分类号
U6 [水路运输]; P75 [海洋工程];
学科分类号
0814 ; 081505 ; 0824 ; 082401 ;
摘要
Currently, deep learning is extensively utilized for ship target detection; however, achieving accurate and real-time detection of multi-scale targets remains a significant challenge. Considering the diverse scenes, varied scales, and complex backgrounds of ships in optical remote sensing images, we introduce a network model named YOLO-RSA. The model consists of a backbone feature extraction network, a multi-scale feature pyramid, and a rotated detection head. We conduct thorough tests on the HRSC2016 and DOTA datasets to validate the proposed algorithm. Through ablation experiments, we assess the impact of each improvement component on the model. In comparative experiments, the proposed model surpasses other models in terms of Recall, Precision, and MAP on the HRSC2016 dataset. Finally, in generalization experiments, our proposed ship detection model exhibits excellent detection performance across various scenarios. The method can accurately detect multi-scale ships in the image and provide a basis for marine ship monitoring and port management.
引用
收藏
页数:20
相关论文
共 53 条
[11]  
Girshick Ross B., 2011, Neural Information Processing Systems, P442
[12]  
He KM, 2020, IEEE T PATTERN ANAL, V42, P386, DOI [10.1109/TPAMI.2018.2844175, 10.1109/ICCV.2017.322]
[13]   Spatial Pyramid Pooling in Deep Convolutional Networks for Visual Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2015, 37 (09) :1904-1916
[14]   Bidirectional mutual guidance transformer for salient object detection in optical remote sensing images [J].
Huang, Kan ;
Tian, Chunwei ;
Li, Ge .
INTERNATIONAL JOURNAL OF REMOTE SENSING, 2023, 44 (13) :4016-4033
[15]  
Jiang YY, 2017, Arxiv, DOI arXiv:1706.09579
[16]   A Densely Connected End-to-End Neural Network for Multiscale and Multiscene SAI Ship Detection [J].
Jiao, Jiao ;
Zhang, Yue ;
Sun, Hao ;
Yang, Xue ;
Gao, Xun ;
Hong, Wen ;
Fu, Kun ;
Sun, Xian .
IEEE ACCESS, 2018, 6 :20881-20892
[17]   Deep Transfer Learning Based Multiway Feature Pyramid Network for Object Detection in Images [J].
Kaur, Parvinder ;
Khehra, Baljit Singh ;
Pharwaha, Amar Partap Singh .
MATHEMATICAL PROBLEMS IN ENGINEERING, 2021, 2021
[18]   Probabilistic Ship Detection and Classification Using Deep Learning [J].
Kim, Kwanghyun ;
Hong, Sungjun ;
Choi, Baehoon ;
Kim, Euntai .
APPLIED SCIENCES-BASEL, 2018, 8 (06)
[19]   Small Infrared Target Detection by Region-Adaptive Clutter Rejection for Sea-Based Infrared Search and Track [J].
Kim, Sungho ;
Lee, Joohyoung .
SENSORS, 2014, 14 (07) :13210-13242
[20]  
Kim T.S., 2019, P IGARSS 2019 2019 I