Improved Ship Target Detection Method for RFBnet Network

doi:10.3778/j.issn.1002-8331.2012-0089

Abstract

Abstract: In order to solve the problem of ship target missing and classification error, a new method of natural image target detection based on improved RFBnet（I-RFBNET） is proposed. Deconvolution feature fusion module and pooling feature fusion module are used for feature fusion to form six new effective feature layers. A step long convolutions method is proposed to extract the concerned area information of the feature unit in the original image. The dilate convolutions block, which integrates the attention mechanism, and the new first three effective feature layers are designed for feature fusion. Focusing classification loss function is introduced to solve the problem of unbalanced distribution of positive and negative samples during training. The model of SeaShips is saved after training SeaShips, a data set of sizing ships. The experimental results show that the improved algorithm has a good detection effect, especially for small targets under multi-target occlusion. The average accuracy is 96.26%, 4.74 percentage points higher than the previous algorithm, and the speed reaches 26 FPS （frame per second）, meeting the need for real-time detection.

Key words: ship detection, attention mechanism, expansion convolution module, feature fusion, small target

摘要： 针对目前舰船目标检测中，多目标情况下的舰船目标很容易被多目标遮挡，造成舰船目标漏检、分类错误等问题，提出了一种基于改进RFBnet（I-RFBnet）的自然图像目标检测方法。使用池化特征融合模块（PFF）和反卷积特征融合模块（DFF）进行特征融合，形成新的六个有效特征层。提出一种跨步长卷积方式来提取特征单元在原图中的关心区域信息，设计了融入注意力机制的膨胀卷积模块（dilate convolutions block，DB）和新的前三个有效特征层再次进行特征融合。引入聚焦分类损失函数解决训练过程中正负样本分布不均衡的问题；最后通过对规模船只检测数据集SeaShips训练后，保存其模型。实验结果表明：改进后的算法检测效果良好，尤其在多目标遮挡下的小目标效果显著。平均精度均值为96.26%，比改进前的算法提高了4.74个百分点，帧率达到26?FPS（frame per second），满足实时检测的需求。

关键词: 舰船检测, 注意力机制, 膨胀卷积模块, 特征融合, 小目标

FANG Jian, LIU Kun. Improved Ship Target Detection Method for RFBnet Network[J]. Computer Engineering and Applications, 2022, 58(12): 155-162.

方健, 刘坤. 改进RFBnet网络的船只目标检测方法[J]. 计算机工程与应用, 2022, 58(12): 155-162.

References

[1] NOVAK L M，HESSE S R.On the performance of orderstatistics CFAR detectors[C]//Proceedings of IEEE 25th Asilomar Conference on Signals，Systems and Computers.Pacific Grove，CA，USA：IEEE，1991：835-840.
[2] GANDHI P P，KASSAM S A.Analysis of CFAR processors in homogeneous background[J].IEEE Transactions on Aerospace & Electronic Systems，2002，24（4）：427-445.
[3] KRIZHEVSKY A，SUTSKEVER I，HINTON G E.Imagenet classification with deep convolutional neural networks[C]//Advances in Neural Information Processing Systems，Lake Tahoe，NV，USA，2012：1097-1105.
[4] GIRSHICK R.Fast R-CNN[C]//Proceedings of the IEEE International Conference on Computer Vision，Boston，MA，USA，2015：1440-1448.
[5] REN S，HE K，GIRSHICK R，et al.Faster R-CNN：towards real time object detection with region proposal networks[C]//IEEE Transactions on Pattern Analysis & Machine Intelligence，2015，39（6）：1137-1149.
[6] REZATOFIGHI H，TSOI N，GWAK J Y，et al.Generalized intersection over union：a metric and a loss for bounding box regression[C]//2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition（CVPR），Long Beach，CA，USA，2019：658-666.
[7] REDMON J，DIVVALA S，GIRSHICK R，et al.You only look once：unified，real-time object detection[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition，Boston，MA，USA，2016：779-788.
[8] LIU W，ANGUELOV D，ERHAN D，et al.SSD： single shot multibox detector[C]//European Conference on Computer Vision，Amsterdam，The Netherlands，2016：21-37.
[9] WANG Y，YANG Y，YAO Y.Single shot multibox detector for ships detection in inland waterway[J].Journal of Harbin Engineering University，2019，40（7）：1258-1262.
[10] 刘相云，郭呈渊，龚志辉，等.一种改进的RFB Net遥感影像目标识别算法[J].测绘科学技术学报，2019，36（2）：179-184.
LIU X Y，GUO C Y，GONG Z H，et al.Object detection in remote sensing image with improved RFB net[J].Journal of Geomatics Science and Technology，2019，36（2）：179-184.
[11] YU Y，LI S，CHEN L，et al.Ship target detection based on improved YOLO v2[J].Journal of Computer Science，2019，46（8）：332-336.
[12] SZEGEDY C，IOFFE S，VANHOUCKE V，et al.Inception-v4，inception-resnet and the impact of residual connections on learning[C]//Thirty-First AAAI Conference on Artificial Intelligence，2017：4278-4284.
[13] 吴文静.带约束的卷积神经网络监控视频船只目标检测[D].武汉：武汉大学，2019.
WU W J.Constrained convolutional neural network for ship target detection in surveillance video[D].Wuhan：Wuhan University，2019.
[14] LIU S T，HUANG D，WANG Y H.Receptive field block net for accurate and fast object detection[J].arXiv：1711.
07767，2017.
[15] LIN Z，JI K，LENG X，et al.Squeeze and excitation rank faster R-CNN for ship detection in SAR images[J].IEEE Geoscience and Remote Sensing Letters，2019，16（5）：751-755.
[16] 刘峰，沈同圣，马新星.特征融合的卷积神经网络多波段舰船目标识别[J].光学学报，2017，37（10）：240-248.
LIU F，SHEN T S，MA X X.Convolutional neural network based multi-band ship target recognition with feature fusion[J].Acta Optica Sinica，2017，37（10）：240-248.
[17] LIN T Y，GOYAL P，GIRSHICK R，et al.Focal loss for dense object detection[C]//Proceedings of the IEEE International Conference on Computer Vision，Venice，Italy，2017：2980-2988.
[18] LI Z，ZHOU F.FSSD：feature fusion single shot multibox detector[J].Computer Vision and Pattern Recognition，2018，36（7）：356-366.
[19] LI X，WANG S.Object detection using convolutional neural networks in a coarse-to-fine manner[J].IEEE Geoscience and Remote Sensing Letters，2017，14（11）：2037-2041.
[20] FU C Y，LIU W，RANGA A，et al.DSSD：deconvolutional singleshot detector[J].arXiv：1701.06659，2017.