Computer Engineering and Applications ›› 2022, Vol. 58 ›› Issue (4): 177-185.DOI: 10.3778/j.issn.1002-8331.2008-0242

• Pattern Recognition and Artificial Intelligence • Previous Articles     Next Articles

Research on Marine Ship Detection Based on Multi-scale Feature Fusion and DCA

PAN Hui, DUAN Xianhua, LUO Binqiang   

  1. School of Computer Science, Jiangsu University of Science and Technology, Zhenjiang, Jiangsu 212003, China
  • Online:2022-02-15 Published:2022-02-15

多尺度特征DCA融合的海上船舶检测算法研究

潘慧,段先华,罗斌强   

  1. 江苏科技大学 计算机学院,江苏 镇江 212003

Abstract: In order to enhance the safety of maritime traffic, an improved ship detection algorithm MS-YOLOV3 is proposed to solve the problems of detection based on YOLOV3 on the ship data set, such as the false detection of target frame and the omission of small targets and low detection accuracy. Firstly, the ship image data set is constructed, including data collection, enhancement and label labeling. The dimension clustering algorithm is used to find the anchor box of appropriate size in the data set and applied to the corresponding scale feature map. Secondly, the model extracts network features based on Darknet-53 and increases network prediction scale. The DCA feature fusion strategy is added to maximize the correlation between high-level features and low-level features, and then the channel splicing is carried out. Measures as said above really improve the detection ability of the model. Finally, GIOU is used as both the parameter of the border optimization function and the border screening index to improve the accuracy of the network in predicting the position information of the border box. The experimental results of MS-YOLOV3 and YOLOV3 show that the accuracy of the former is improved by 7.9 percentage points. At the same time, the added loss of GIOU frame reduces the average loss of the model, strengthens the robustness of the model, and greatly reduces the positioning error of the target frame. According to the training effect on Pascal VOC2007 data set, the average accuracy of MS-YOLOV3 is better than the series algorithm of YOLO, SSD300 and Faster-RCNN. The proposed feature interactive detection model in this paper makes the ship position information and classification more accurate.

Key words: deep learning, convolutional neural network(CNN), ship detection, multiscale features, YOLOV3, discriminant correlation analysis(DCA);generalized intersection over union(GIOU)

摘要: 为了加强海上交通的安全性,以常见的民用船和军用船为研究对象,针对原始YOLOV3算法在船舶数据集上检测精度不高、目标框出现误检和小目标漏检的问题,提出了改进的船舶检测算法MS-YOLOV3。构建船舶图像数据集Shipdataset,包括数据采集、增强和标签标注,使用维度聚类算法在该数据集中找出合适尺寸的先验框,并应用于相对应的尺度特征图。以Darknet-53的网络框架为基础特征提取网络,增加网络预测尺度,在多尺度特征融合中加入DCA融合策略,提高模型对船舶的检测能力。以MS-YOLOV3为算法框架,采用GIOU作为边框损失函数的参数,提升模型对边界框位置信息的预测准确度。结果MS-YOLOV3与YOLOV3检测算法的对比实验表明,前者在船舶数据集上的精度有7.9个百分点的提升。同时加入的GIOU边框损失,拉低了模型的平均损失,加强了模型的鲁棒性,使得目标框的定位误差大大减小。根据Pascal VOC2007数据集上的训练效果,MS-YOLOV3的平均精度相较于YOLO系列算法、SSD300和Faster-RCNN,精确度优势更加明显。提出的MS-YOLOV3检测模型使得船舶的位置信息和类别精度更加准确。

关键词: 深度学习, 卷积神经网络(CNN), 船舶检测, 多尺度特征, YOLOV3, 判别相关分析(DCA), 广义交并比(GIOU)