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 条
[1]   Speeded-Up Robust Features (SURF) [J].
Bay, Herbert ;
Ess, Andreas ;
Tuytelaars, Tinne ;
Van Gool, Luc .
COMPUTER VISION AND IMAGE UNDERSTANDING, 2008, 110 (03) :346-359
[2]  
Bousetouane F., 2016, P 2016 13 IEEE INT C
[3]   Remote Sensing Image Registration Based on Modified SIFT and Feature Slope Grouping [J].
Chang, Herng-Hua ;
Wu, Guan-Long ;
Chiang, Mao-Hsiung .
IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2019, 16 (09) :1363-1367
[4]   Multi-Scale Ship Detection Algorithm Based on YOLOv7 for Complex Scene SAR Images [J].
Chen, Zhuo ;
Liu, Chang ;
Filaretov, V. F. ;
Yukhimets, D. A. .
REMOTE SENSING, 2023, 15 (08)
[5]   A survey on object detection in optical remote sensing images [J].
Cheng, Gong ;
Han, Junwei .
ISPRS JOURNAL OF PHOTOGRAMMETRY AND REMOTE SENSING, 2016, 117 :11-28
[6]  
Colgan C.S., 2018, The Blue Economy Handbook of the Indian Ocean Region, V38
[7]   Histograms of oriented gradients for human detection [J].
Dalal, N ;
Triggs, B .
2005 IEEE COMPUTER SOCIETY CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, VOL 1, PROCEEDINGS, 2005, :886-893
[8]   CenterNet: Keypoint Triplets for Object Detection [J].
Duan, Kaiwen ;
Bai, Song ;
Xie, Lingxi ;
Qi, Honggang ;
Huang, Qingming ;
Tian, Qi .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :6568-6577
[9]   Fast R-CNN [J].
Girshick, Ross .
2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, :1440-1448
[10]   Rich feature hierarchies for accurate object detection and semantic segmentation [J].
Girshick, Ross ;
Donahue, Jeff ;
Darrell, Trevor ;
Malik, Jitendra .
2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, :580-587