高级检索
乔美英, 史建柯, 李冰锋, 赵岩, 史有强. 改进损失函数的增强型FPN水下小目标检测[J]. 计算机辅助设计与图形学学报, 2023, 35(4): 525-537. DOI: 10.3724/SP.J.1089.2023.19381
引用本文: 乔美英, 史建柯, 李冰锋, 赵岩, 史有强. 改进损失函数的增强型FPN水下小目标检测[J]. 计算机辅助设计与图形学学报, 2023, 35(4): 525-537. DOI: 10.3724/SP.J.1089.2023.19381
Qiao Meiying, Shi Jianke, Li Bingfeng, Zhao Yan, and Shi Youqiang. Enhanced FPN Underwater Small Target Detection with Improved Loss Function[J]. Journal of Computer-Aided Design & Computer Graphics, 2023, 35(4): 525-537. DOI: 10.3724/SP.J.1089.2023.19381
Citation: Qiao Meiying, Shi Jianke, Li Bingfeng, Zhao Yan, and Shi Youqiang. Enhanced FPN Underwater Small Target Detection with Improved Loss Function[J]. Journal of Computer-Aided Design & Computer Graphics, 2023, 35(4): 525-537. DOI: 10.3724/SP.J.1089.2023.19381

改进损失函数的增强型FPN水下小目标检测

Enhanced FPN Underwater Small Target Detection with Improved Loss Function

  • 摘要: 针对水下小目标因携带特征信息少、定位不精准而导致检测精度低的问题,提出一种特征金字塔网络(FPN).首先,在FPN上采样过程中加入协同非局部注意力模块,利用卷积、横纵向池化挖掘特征图的静态和动态上下文信息;其次,在FPN通道调整过程中加入三叉戟特征增强模块,利用并行空洞卷积与高效通道注意力(ECANet)捕捉多尺度空间与通道特征信息;最后,在Faster R-CNN算法的回归损失函数中引入线性回归损失增益系数,增大对多尺度目标回归偏移量的惩罚,提高定位精度.实验结果表明,采用2020年全国水下目标检测大赛提供的数据集、PASCAL VOC数据集和MS COCO数据集进行实验,该算法比基线Faster R-CNN算法精度分别提升2.8%,2.2%和2.5%,结果证明了其有效性.

     

    Abstract: A feature pyramid network (FPN) is proposed, which resolves low detection accuracy of underwater small target, due to lack of feature information and inaccurate positioning. Firstly, convolution, horizontal and vertical pooling are used to extract static and dynamic context information of feature maps, which is named coordinative nonlocal attention module and plugged into upsampling process of FPN. Secondly, parallel dilate convolution and efficient channel attention network (ECANet) are used to capture multi-scale space and channel feature information, which is named trident feature enhancement module and plugged into channel adjustment process of FPN. Finally, the linear regression loss gain coefficient is introduced into the regression loss function of the Faster R-CNN algorithm, to increase the penalty for the multi-scale target regression offset and improve the positioning accuracy. The experimental results show that, compared with the baseline Faster R-CNN algorithm, the proposed algorithm improves the detection accuracy by 2.8%, 2.2% and 2.5% in the dataset provided by the 2020 National Underwater Target Detection Contest, PASCAL VOC dataset and the MS COCO dataset, respectively, which proves the effectiveness of the method.

     

/

返回文章
返回