计算机科学
水下
目标检测
算法
计算机视觉
人工智能
模式识别(心理学)
地质学
海洋学
作者
Jinghua Huang,Chao Fang,Xiaogang Zheng,Jue Liu
出处
期刊:IEEE Access
[Institute of Electrical and Electronics Engineers]
日期:2024-01-01
卷期号:12: 172186-172195
被引量:6
标识
DOI:10.1109/access.2024.3496925
摘要
Underwater object detection technology is widely used in fields such as ocean exploration. However, due to the complex underwater environment, issues like light attenuation and scattering lead to low detection accuracy, which fails to meet the requirements. To address these issues, we propose an improved YOLOv8n-based model called YOLOv8-UC. This model incorporates a modified Dilation-wise Residual (DWR) C2f module to enhance the ability to extract features from the network’s high-level expandable receptive fields. It also integrates the Large Separable Kernel Attention (LSKA) module with the SPPF of YOLOv8 to enhance multi-scale feature extraction capabilities, reducing the loss of details. To solve the problem of redundant parameters and computational load in the detection head, the original detection head is replaced with a shared parameter structure, and RepConv is introduced. Additionally, the Inner-SIoU loss function is improved by using auxiliary boundaries at different scales to accelerate bounding box regression and improve detection accuracy. Experimental results show that the designed YOLOv8-UC achieves an mAP@0.5 of 79.3%, with a 6.9% increase in detection accuracy (P) and a 5.9% increase in precision (mAP@0.5) compared to YOLOv8n, demonstrating the effectiveness and application prospects of this method.
科研通智能强力驱动
Strongly Powered by AbleSci AI