基于改进YOLOv5s的海下垃圾检测算法  

Detection Algorithoy of Undersea Garbage Based on Improved YOLOv5s

作　　者：吴观茂[1] 王涛 WU Guanmao;WANG Tao(College of Computer Science and Engineering,Anhui University of Science and Technology,Huainan Anhui 232001)

机构地区：[1]安徽理工大学计算机科学与工程学院,安徽淮南232001

出　　处：《湖北理工学院学报》2024年第1期47-51,共5页Journal of Hubei Polytechnic University

基　　金：安徽省自然科学基金面上项目(项目编号:1908085MF189);安徽省重点研究与开发计划项目(项目编号:202004b11020029)。

摘　　要：为了提升海下垃圾检测精度,降低漏检频率,提出了一种基于改进YOLOv5s的检测算法。该算法添加了ECA模块来关注感兴趣区域,构造了四级多尺度特征融合使提取的特征图信息更加丰富,引用损失函数使得边框回归更加合适与准确。改进的YOLOv5s算法在TrashCan数据集上的mAP由原来的87.3%提高到92.1%,具有较高的精度。In order to improve the accuracy of undersea garbage detection and reduce the occurrence of missed detection,this paper proposes a detection algorithm based on improved YOLOv5s.The algorithm adds an ECA module to focus on the region of interest,constructs a four-level multi-scale feature fusion to make the extracted feature map more informative,and references the loss function to make the edge regression more appropriate and accurate.The improved YOLOv5s algorithm improves the mAP on the TrashCan dataset from 87.3%to 92.1%with high accuracy.

关 键 词：海下垃圾 深度学习 小目标 多尺度融合 EIoU 

分 类 号：TP391[自动化与计算机技术—计算机应用技术]