首页|基于改进YOLOv5的海洋生物检测算法

基于改进YOLOv5的海洋生物检测算法

扫码查看
原文链接
  • 万方数据
对于水下机器人捕捞行业,提高对海洋生物的识别准确率可以有效减少对海洋生态环境的破坏.针对现有海洋生物检测模型在复杂环境下对小目标存在特征提取能力不足、检测精度低的问题,提出了一种基于改进YOLOv5的海洋生物检测算法.该算法通过改进马赛克数据增强,生成更多的小目标数据样本;在YOLOv5的主干网络引入SimAM(Simple,Parameter-Free Attention Module)无参注意力机制,该注意力机制对海洋生物特征图分配3D注意力权值,从而增强模型提取特征的能力.实验结果表明,对比原始YOLOv5算法,在没有引入额外参数的情况下,查准率、查全率、平均检测精度分别提高了2.8％、1.7％、1.6％,为水下机器人进行精准识别和捕捞提供了技术支持.
Marine Organism Detection Algorithm Based on Improved YOLOv5

朱伟东、何月顺、陈杰、任维民、孙一蓬

展开 >

东华理工大学信息工程学院 南昌 330013

江西经济管理干部学院 南昌 330088

海洋生物 YOLOv5 马赛克数据增强 SimAM

41872243

2022

10.3969/j.issn.1672-9722.2022.08.001

计算机与数字工程
中国船舶重工集团公司第七0九研究所

计算机与数字工程

CSTPCD
影响因子：0.355
ISSN：1672-9722
年,卷(期)：2022.50(8)
  • 7