改进YOLOv11n的水面垃圾检测算法
DOI:
CSTR:
作者:
作者单位:

1.华北理工大学电气工程学院 唐山 063200; 2.华北理工大学招生就业处 唐山 063200

作者简介:

通讯作者:

中图分类号:

TN911.73;TN919.8

基金项目:

河北省自然科学基金(D2024209006)、河北省教育厅科学研究项目(QN2024147)资助


Improved YOLOv11n-based water surface garbage detection algorithm
Author:
Affiliation:

1.School of Electrical Engineering, North China University of Science and Technology,Tangshan 063200, China; 2.Admissions and Employment Office, North China University of Science and Technology,Tangshan 063200, China

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    为应对水波干扰、光线反射及漂浮物遮挡等复杂水面背景下目标检测精度下降的问题,提出一种改进YOLOv11n的水面垃圾检测算法。首先,利用FCM模块对C3k2中的Bottleneck进行重新设计,构建C3k2-FCM模块,缓解下采样过程中物体空间位置信息丢失的问题;其次,引入具有不同膨胀率的共享卷积层并结合空间与通道协同注意力机制SCSA,构建FPSC-SCSA模块取代SPPF,增加对关键区域的关注能力,减少关键信息的丢失;再次,将颈部聚合网络替换为多尺度特征融合网络并嵌入U-Net V2的SDI模块构建BIFPN-V2S模块,扩展模型感受野并强化全局上下文信息交互能力;最后,构建SGSV-IoU损失函数,提高模型对边界形状和细节的刻画精度,优化对不规则漂浮物的定位效果。实验结果表明:改进后的YOLOv11n算法在自建水面垃圾数据集上相较原算法mAP50提高了2.8%,参数量和计算量分别减少了0.81 M和0.1 GFLOPs,证明了改进算法的有效性。

    Abstract:

    To address the decline in detection accuracy caused by complex water surface backgrounds such as wave interference, light reflection, and object occlusion, an improved YOLOv11n-based water surface garbage detection algorithm is proposed. First, the Bottleneck structure in C3k2 is replaced with the FCM module to construct the C3k2-FCM module, mitigating the loss of spatial position information during downsampling. Second, a FPSC-SCSA module is designed by introducing shared convolutional layers with different dilation rates and the SCSA mechanism to replace the SPPF module, enhancing the model′s focus on key regions and reducing the loss of crucial information. Third, a BIFPN-V2S module is developed by replacing the neck aggregation network with a multi-scale feature fusion network embedded with the SDI module from U-Net V2, expanding the receptive field and strengthening global contextual interactions. Finally, an SGSV-IoU loss function is designed to improve boundary shape and detail representation, enhancing the localization of irregular floating objects. Experimental results on a self-built water surface garbage dataset show that, compared with the original YOLOv11n model, the improved algorithm increases mAP50 by 2.8%, while reducing parameters and computation by 0.81 M and 0.1 GFLOPs, respectively, demonstrating the effectiveness of the proposed method.

    参考文献
    相似文献
    引证文献
引用本文

王海群,谢伟民,晁帅,于海峰.改进YOLOv11n的水面垃圾检测算法[J].电子测量技术,2026,49(9):110-120

复制
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2026-06-08
  • 出版日期:
文章二维码

重要通知公告

①《电子测量技术》期刊收款账户变更公告