基于轻量化网络的海底垃圾检测算法SLD-Net

作     者：周华平 汪佳伟 ZHOU Huaping;WANG Jiawei

作者机构：安徽理工大学计算机科学与工程学院安徽淮南232001 

出 版 物：《湖北理工学院学报》 (Journal of Hubei Polytechnic University)

年 卷 期：2024年第40卷第1期

页      码：41-46页

摘      要：设计了一种轻量级的两阶段海洋垃圾检测算法SLD-Net(Seabed Litter Detection Network)。一方面,在预处理阶段构建了一个由颜色转换子模块和图像去噪子模块组成的图像增强模块MDIE,提高了图像质量;另一方面,采用轻量化的MobilenetV2作为目标检测的骨干网络,并引入一种改进的双路FPN结构,对深层特征图进行增强,提高对小目标的检测能力。经图像增强和目标检测网络联合训练后,进一步提升了模型的精度。在J-EDI海洋垃圾数据集上进行实验验证,mAP和速度分别达到了94.5%和65 FPS,且模型参数量仅有5.4 M,表明SLD-Net算法在精度、速度和参数量上达到了很好的效果。

主 题 词：神经网络 海底垃圾检测 轻量化网络 图像增强 

学科分类：08[工学] 080203[080203] 0802[工学-机械学] 

D　O　I：10.3969/j.issn.2095-4565.2024.01.008

馆 藏 号：203125483...