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