基于改进YOLOv5的海洋生物检测算法
朱伟东1
何月顺1
陈杰1
任维民1
孙一蓬2
1.东华理工大学信息工程学院 南昌 3300132.江西经济管理干部学院 南昌 330088
摘要：对于水下机器人捕捞行业,提高对海洋生物的识别准确率可以有效减少对海洋生态环境的破坏.针对现有海洋生物检测模型在复杂环境下对小目标存在特征提取能力不足、检测精度低的问题,提出了一种基于改进YOLOv5的海洋生物检测算法.该算法通过改进马赛克数据增强,生成更多的小目标数据样本;在YOLOv5的主干网络引入SimAM(Simple,Parameter-Free Attention Module)无参注意力机制,该注意力机制对海洋生物特征图分配3D注意力权值,从而增强模型提取特征的能力.实验结果表明,对比原始YOLOv5算法,在没有引入额外参数的情况下,查准率、查全率、平均检测精度分别提高了2.8％、1.7％、1.6％,为水下机器人进行精准识别和捕捞提供了技术支持.
关键词：海洋生物YOLOv5马赛克数据增强SimAM
分类号：TP391.41(计算技术、计算机技术)
资助基金：国家自然科学基金(41872243)
论文发表日期：2022-08-20
在线出版日期：2025-08-15（本平台首次上网日期，不代表文献的发表时间）
页数：6( 1631-1636 )
英文信息