Study on Visual Detection Algorithm of Sea Surface Targets Based on Improved YOLOv3

被引:30
作者
Liu, Tao [1 ]
Pang, Bo [1 ]
Ai, Shangmao [2 ]
Sun, Xiaoqiang [1 ]
机构
[1] Harbin Engn Univ, Coll Intelligent Syst Sci & Engn, Harbin 150001, Peoples R China
[2] Harbin Engn Univ, Coll Shipbldg Engn, Harbin 150001, Peoples R China
基金
中国国家自然科学基金;
关键词
YOLO v3; anchor-setting; target detection; feature fusion; connection of cross-feature maps; buoys; ships;
D O I
10.3390/s20247263
中图分类号
O65 [分析化学];
学科分类号
070302 ; 081704 ;
摘要
Countries around the world have paid increasing attention to the issue of marine security, and sea target detection is a key task to ensure marine safety. Therefore, it is of great significance to propose an efficient and accurate sea-surface target detection algorithm. The anchor-setting method of the traditional YOLO v3 only uses the degree of overlap between the anchor and the ground-truth box as the standard. As a result, the information of some feature maps cannot be used, and the required accuracy of target detection is hard to achieve in a complex sea environment. Therefore, two new anchor-setting methods for the visual detection of sea targets were proposed in this paper: the average method and the select-all method. In addition, cross PANet, a feature fusion structure for cross-feature maps was developed and was used to obtain a better baseline cross YOLO v3, where different anchor-setting methods were combined with a focal loss for experimental comparison in the datasets of sea buoys and existing sea ships, SeaBuoys and SeaShips, respectively. The results showed that the method proposed in this paper could significantly improve the accuracy of YOLO v3 in detecting sea-surface targets, and the highest value of mAP in the two datasets is 98.37% and 90.58%, respectively.
引用
收藏
页码:1 / 14
页数:14
相关论文
共 35 条
[1]   Research on Airplane and Ship Detection of Aerial Remote Sensing Images Based on Convolutional Neural Network [J].
Cao, Changqing ;
Wu, Jin ;
Zeng, Xiaodong ;
Feng, Zhejun ;
Wang, Ting ;
Yan, Xu ;
Wu, Zengyan ;
Wu, Qifan ;
Huang, Ziqiang .
SENSORS, 2020, 20 (17) :1-16
[2]   Deep learning for autonomous ship-oriented small ship detection [J].
Chen, Zhijun ;
Chen, Depeng ;
Zhang, Yishi ;
Cheng, Xiaozhao ;
Zhang, Mingyang ;
Wu, Chaozhong .
SAFETY SCIENCE, 2020, 130
[3]  
Cui H., 2019, OCEANS-IEEE
[4]   A Novel Detector Based on Convolution Neural Networks for Multiscale SAR Ship Detection in Complex Background [J].
Dai, Wenxin ;
Mao, Yuqing ;
Yuan, Rongao ;
Liu, Yijing ;
Pu, Xuemei ;
Li, Chuan .
SENSORS, 2020, 20 (09)
[5]   Learning a robust CNN-based rotation insensitive model for ship detection in VHR remote sensing images [J].
Dong, Zhong ;
Lin, Baojun .
INTERNATIONAL JOURNAL OF REMOTE SENSING, 2020, 41 (09) :3614-3626
[6]   Fast R-CNN [J].
Girshick, Ross .
2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, :1440-1448
[7]   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
[8]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[9]  
He T., 2019, P 2019 IEEE CVF C CO
[10]   Densely Connected Convolutional Networks [J].
Huang, Gao ;
Liu, Zhuang ;
van der Maaten, Laurens ;
Weinberger, Kilian Q. .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :2261-2269