| 注册
首页|期刊导航|重庆工商大学学报(自然科学版)|基于SimAM注意力机制的DCN-YOLOv5水下目标检测

基于SimAM注意力机制的DCN-YOLOv5水下目标检测

刘向举 刘洋 蒋社想

重庆工商大学学报(自然科学版)2025,Vol.42Issue(2):63-70,8.
重庆工商大学学报(自然科学版)2025,Vol.42Issue(2):63-70,8.DOI:10.16055/j.issn.1672-058X.2025.0002.009

基于SimAM注意力机制的DCN-YOLOv5水下目标检测

DCN-YOLOv5 Underwater Target Detection Based on SimAM Attention Mechanism

刘向举 1刘洋 1蒋社想1

作者信息

  • 1. 安徽理工大学计算机科学与工程学院,安徽淮南 232001
  • 折叠

摘要

Abstract

Objective Given the complex underwater environment,the target boundary may be blurred or the appearance and shape of the underwater target may be non-rigidly deformed due to light refraction,which makes underwater target detection difficult.A DCN-YOLOv5 underwater target detection method based on the SimAM attention mechanism was proposed.Methods Firstly,the bi-directional feature pyramid network(BiFPN)used by YOLOv5 was used to extract and fuse feature information on multiple scales to improve the accuracy of target recognition.Secondly,to address the variations in appearance and shape of underwater objects,the CBS module in the C3 module was combined with the deformable convolution network(DCN),and the DBS module was proposed.The DBS module was used to form the D3 module and replace part of the C3 module to adapt to the changing appearance and shape of the underwater targets.At the same time,the weighted attention mechanism was integrated to adaptively adjust the attention of the model and further improve the feature expression ability in complex scenes.Finally,considering the fuzzy boundary of the target and to improve the target positioning accuracy,the WIoU(Wise-IoU)loss function was used to replace the cross-entropy loss,which can better adapt to the characteristics of different target types and sizes and improve the robustness of the algorithm.Results Experimental results showed that DCN-YOLOv5 achieved an average precision(mAP)of 87.57%,outperforming YOLOv5 and other classical networks,with an average identification time of only 24.5 ms per image.Conclusion The experimental results demonstrate that the model significantly improves detection accuracy while ensuring real-time detection,providing valuable insights for the practical use of underwater target detection.

关键词

水下目标检测/SimAM注意力机制/可变形卷积/WIoU

Key words

underwater target detection/SimAM attention mechanism/deformable convolutions/WioU

分类

计算机与自动化

引用本文复制引用

刘向举,刘洋,蒋社想..基于SimAM注意力机制的DCN-YOLOv5水下目标检测[J].重庆工商大学学报(自然科学版),2025,42(2):63-70,8.

基金项目

安徽省重点实验室项目(ZKSYS202204). (ZKSYS202204)

重庆工商大学学报(自然科学版)

1672-058X

访问量0
|
下载量0
段落导航相关论文