黄廷辉, 高新宇, 黄春德, 何曰平. 基于FAttention-YOLOv5的水下目标检测算法研究[J]. 微电子学与计算机, 2022, 39(6): 60-68. DOI: 10.19304/J.ISSN1000-7180.2021.1261
引用本文: 黄廷辉, 高新宇, 黄春德, 何曰平. 基于FAttention-YOLOv5的水下目标检测算法研究[J]. 微电子学与计算机, 2022, 39(6): 60-68. DOI: 10.19304/J.ISSN1000-7180.2021.1261
HUANG Tinghui, GAO Xinyu, HUANG Chunde, HE Yueping. Research on underwater target detection algorithm based on FAttention-YOLOv5[J]. Microelectronics & Computer, 2022, 39(6): 60-68. DOI: 10.19304/J.ISSN1000-7180.2021.1261
Citation: HUANG Tinghui, GAO Xinyu, HUANG Chunde, HE Yueping. Research on underwater target detection algorithm based on FAttention-YOLOv5[J]. Microelectronics & Computer, 2022, 39(6): 60-68. DOI: 10.19304/J.ISSN1000-7180.2021.1261

基于FAttention-YOLOv5的水下目标检测算法研究

Research on underwater target detection algorithm based on FAttention-YOLOv5

  • 摘要: 针对目前现有算法不能很好适用于水下目标检测,同时为提高水下目标检测的实时性和准确性,提出一种基于F-CBAM注意力机制的YOLOv5水下目标检测网络模型FAttention-YOLOv5.模型采用单阶段目标检测网络模型YOLOv5作为基础模型,在模型中嵌入提出的F-CBAM注意力机制,通过在CBAM结构中引用FReLU激活函数,在激活函数阶段通过二维空间捕捉复杂的特征分布情况,实现像素级的空间信息建模能力,提高模型准确率;采用F-CBAM中的通道注意力机制和空间注意力机制提高目标物体的通道权重以及扩大目标对原图的感受野,提高目标检测模型对特征的学习能力;并在FAttention-YOLOv5模型中融合递归网络特征金字塔,通过特征递归使网络充分学习不同尺度的图像特征,从而提高小目标的检测精度;最后对改进模型的损失函数进行优化,避免新模型梯度消失或爆炸.实验结果表明:所设计的水下目标检测模型FAttention-YOLOv5,可以提高模型的特征提取能力,从而有效提高水下目标检测的准确度,为海洋生物捕捉提供新型解决方案和技术辅助.

     

    Abstract: In order to improve the real-time and accuracy of underwater target detection, an underwater target detection network model FAttention-YOLOv5 based on F-CBAM attention mechanism is proposed. The model uses the single-stage target detection network model YOLOv5 as the basic model, embeds the proposed F-CBAM attention mechanism in the model, references the FReLU activation function in the CBAM structure, and captures the complex feature distribution through two-dimensional space in the activation function stage, so as to realize the spatial information modeling ability at the pixel level and improve the accuracy of the model; The channel attention mechanism and spatial attention mechanism in F-CBAM are used to improve the channel weight of the target object and expand the receptive field of the target to the original image, so as to improve the feature learning ability of the target detection model; The recursive network feature pyramid is fused in the FAttention-YOLOv5 model, and the network can fully learn the image features of different scales through feature recursion, so as to improve the detection accuracy of small targets; Finally, the loss function of the improved model is optimized to avoid the disappearance or explosion of the gradient of the new model. The experimental results show that the designed underwater target detection model FAttention-YOLOv5 can improve the feature extraction ability of the model, effectively improve the accuracy of underwater target detection, and provide a new solution and technical assistance for marine biological capture.

     

/

返回文章
返回