计算机时代2023,Issue(10) :120-125.DOI:10.16644/j.cnki.cn33-1094/tp.2023.10.025

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

Marine garbage target detection algorithm based on improved YOLOv5s

刘将 涂振宇 李元汉 李豪
计算机时代2023,Issue(10) :120-125.DOI:10.16644/j.cnki.cn33-1094/tp.2023.10.025

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

Marine garbage target detection algorithm based on improved YOLOv5s

刘将 1涂振宇 1李元汉 1李豪1
扫码查看

作者信息

  • 1. 南昌工程学院信息工程学院,江西 南昌 330099
  • 折叠

摘要

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

关键词

YOLOv5s/海洋垃圾/LKA/SPPF

Key words

YOLOv5s/marine garbage/LKA/SPPF

引用本文复制引用

基金项目

江西省水利厅科技项目(KT201639)

江西省科技厅重点研发计划(20151BBE50077)

出版年

2023
计算机时代
浙江省计算技术研究所 浙江省计算机学会

计算机时代

影响因子:0.411
ISSN:1006-8228
参考文献量4
段落导航相关论文