基于改进YOLOv5s的海洋垃圾目标检测算法

作者：刘将; 涂振宇^*; 李元汉; 李豪

来源：计算机时代, 2023, (10): 120-125.

DOI：10.16644/j.cnki.cn33-1094/tp.2023.10.025

• YOLOv5s
• 海洋垃圾
• LKA
• SPPF

为了减少海洋垃圾对生物和水资源造成的危害，提出一种基于改进YOLOv5s的海洋垃圾目标检测算法。针对目标检测算法中小目标漏检及特征提取能力不足等问题，在主干网络部分添加一种基于大核注意力（LKA）机制的改进模块对特征层进行关键目标提取，并改变了空间金字塔池化（SPPF）网络结构，目的是让深层特征图上的每个像素点在输入图像上映射的区域变大。实验结果表明，改进后的YOLOv5s目标检测模型的平均精度均值（mAP）提升6%。

• 单位
  南昌工程学院