上海海洋大学学报2025，Vol.34Issue(3)：696-706,11.DOI:10.12024/jsou.20240404474
YOLO-U:基于结构重参数化和双重注意力机制的水下目标检测算法
YOLO-U:An underwater object detection algorithm based on structural reparameterization and dual attention mechanism
摘要
Abstract
The research on underwater object detection algorithms is a prerequisite for achieving intelligent fishing with underwater robots.The problems of fuzzy object,numerous small objects and mutual occlusion in underwater object detection pose challenges to the realization of accurate object detection.This paper proposes a YOLO-U algorithm for underwater object detection based on YOLOv7-tiny.The algorithm introduces a RepViT backbone network with structural reparameterization and fuses an ESE channel attention mechanism to enhance the feature extraction capability for underwater fuzzy objects.Additionally,a feature pyramid network CAFPN with shallow coordinate information feature fusion is designed to further enhance the sensitivity of the detection model to directional and positional information,and integrate feature information of different scales to improve the detection ability of small objects.Furthermore,the WIoUv2 bounding box loss function is employed to effectively reduce the contribution of easy examples to the loss value.This allows the model to focus on occluded objects and further improve the detection accuracy for occluded objects.The YOLO-U algorithm achieves a mAP50 of 84.6％on the URPC2021 dataset,which is an improvement of 2.1％,5.2％,and 2.8％compared to YOLOv7-tiny,YOLOv5s,and YOLOv8s,respectively.The detection results show that the algorithm can effectively improve the detection accuracy of underwater objects and further improve the detection performance of underwater fuzzy objects,small objects,and occluded objects.关键词
水下目标检测/YOLOv7-tiny/结构重参数化/注意力机制/损失函数Key words
underwater object detection/YOLOv7-tiny/structural reparameterization/attention mechanism/loss function分类
农业科技引用本文复制引用
李江川,韩彦岭,董传胜,王艳,王静,张云,杨树瑚..YOLO-U:基于结构重参数化和双重注意力机制的水下目标检测算法[J].上海海洋大学学报,2025,34(3):696-706,11.基金项目
国家自然科学基金面上项目(42176175,42271335) （42176175,42271335）
十三五"蓝色粮仓科技创新"国家重点研发计划(2019YFD0900805) （2019YFD0900805）
