兰州工业学院学报2024,Vol.31Issue(5) :1-7,90.

基于YOLOv8n的水下生物目标检测方法

Underwater Biological Target Detection Method Based on YOLOv8n

庞军舰 居锦武 石睿
兰州工业学院学报2024,Vol.31Issue(5) :1-7,90.

基于YOLOv8n的水下生物目标检测方法

Underwater Biological Target Detection Method Based on YOLOv8n

庞军舰 1居锦武 1石睿1
扫码查看

作者信息

  • 1. 四川轻化工大学 计算机科学与工程学院,四川 宜宾 644000
  • 折叠

摘要

针对由于水下环境背景复杂、能见度较低、光照分布不均匀等因素导致网络特征提取不充分的问题,提出了一种基于YOLOv8n改进的YOLOv8n-DSP 水下生物目标检测方法.首先,在YOLOv8n的特征提取主干网络中添加DBB(Diverse Branch Block)模块,将其与C2f模块进行融合,使其组合不同尺度和复杂度的多个分支,丰富特征空间,强化网络的特征提取能力;然后在空间金字塔池化前加入加权注意力机制(SimAM),使模型的关注点自适应调节,增强在复杂场景下的表征能力;最后为进一步提高目标检测框定位精度,将已有的CIoU更换为PIoU(Powerful-IoU),使得目标检测框更好地回归.结果表明:YOLOv8n-DSP 网络相较于原来的YOLOv8n网络在mAP@0.5 和mAP@0.5:0.95 分别提升了2%和2.2%.

Abstract

In response to the insufficient feature extraction caused by factors such as the complex underwater en-vironment,low visibility,and uneven light distribution,a YOLOv8n-DSP underwater biological target detection method based on the improvement of YOLOv8n is proposed.Firstly,a DBB(Diverse Branch Block)module is added to the feature extraction backbone network of YOLOv8n and fused with the C2f module to combine multiple branches of different scales and complexities,enriching the feature space and strengthening the network's feature extraction capability.Then,a weighted attention mechanism(SimAM)is added before spatial pyramid pooling to adaptively adjust the model's focus and enhance its representation ability in complex scenes.Finally,to further improve the localization accuracy of the object detection boxes,the existing CIoU is replaced with PIoU(Power-ful-IoU),resulting in better regression of the object detection boxes.Experimental results show that the YOLOv8n-DSP network improves mAP@0.5 and mAP@0.5:0.95 by 2%and 2.2%,respectively,compared to the original YOLOv8n network.

关键词

YOLOv8/深度学习/水下目标检测/注意力机制/PIoU

Key words

YOLOv8/deep learning/underwater object detection/attention mechanism/PIoU

引用本文复制引用

出版年

2024
兰州工业学院学报
兰州工业学院

兰州工业学院学报

影响因子:0.205
ISSN:1009-2269
段落导航相关论文