一种复杂环境下无人艇目标检测系统及方法技术方案

技术编号:39666999 阅读:10 留言:0更新日期:2023-12-11 18:31
一种复杂环境下无人艇目标检测方法,包括以下步骤:使用

【技术实现步骤摘要】
一种复杂环境下无人艇目标检测系统及方法


[0001]本专利技术涉及陶瓷胚体制作
,特别是一种复杂环境下无人艇目标检测系统及方法


技术介绍

[0002]当前,随着科技与社会需求的快速发展,无人平台的发展也越来越快

基于水上作业的需求,无人艇也得到迅速发展并使用各个水上作业工作中,如水上巡航

清洁

救援等

在复杂的水上环境进行目标识别任务时,目前常用的目标检测算法是
YOLOv5。
[0003]YOLOv5
骨干网络使用的是基于
CSPDarknet53
的骨干网络结构,虽然这些网络在提取单一尺度的特征方面表现良好,但缺少明确的多尺度信息融合机制,这会导致网络对不同尺度目标的检测能力不够强大,缺乏多尺度信息融合的问题

同时,应用于无人艇目标检测时,在复杂场景中存在大量遮挡

尺度变化等因素,这会导致骨干网络的特征提取能力受限,影响目标检测的准确性和鲁棒性

[0004]YOLOv5
中的
PANet
结构旨在改善特征融合过程,将骨干网络不同层级的特征进行融合

它通过侧连接将浅层特征与深层特征相结合,实现了高分辨率特征和低分辨率特征的融合

这使得模型能够同时捕捉到深层特征的语义信息和浅层特征的细节信息


PAnet
需要进行跨阶段的特征聚合和融合,这涉及到大量的特征图操作和通道交互,需要较高的计算资源

这会导致训练和推理的速度较慢,特别是在资源受限的设备上,同时
PAnet
中的跨阶段连接和特征融合需要额外的参数来进行操作

这增加了模型的参数数量,会增加过拟合的风险,并且需要更多的存储空间

此外,
PAnet
结构对小目标处理仍然有局限性,由于水面环境复杂,无人艇容易受到水浪及附近环境干扰,会受到特征分辨率较低的影响,导致识别精度下降,系统鲁棒性较差


技术实现思路

[0005]针对上述缺陷,本专利技术的目的在于提出一种复杂环境下无人艇目标检测系统及方法,在确保目标识别的准确性前提下,提高检测的鲁棒性

[0006]一种复杂环境下无人艇目标检测方法,包括以下步骤:
[0007]使用
LabelImg
对一定数量待检测物体图像进行标注,获得训练的数据集;
[0008]对
YOLOv5s
模型进行改进,其中改进包括将
ECA
注意力机制模块融入
YOLOv5s
模型的
Backbone
以及将
YOLOv5s
模型中将自底向上特征金字塔改为加权双向特征金字塔网络;
[0009]将数据集输入至改进后的
YOLOv5s
模型,对
YOLOv5s
模型进行训练;
[0010]将训练好的
YOLOv5s
模型嵌入至无人艇中,实现目标识别

[0011]优选的,所述将
ECA
注意力机制模块融入
YOLOv5s
模型的
Backbone
前还需要执行如下操作:
[0012]ECA
注意力机制模块在
SE
模块上使用全连接层
FC
学习通道注意信息,并使用卷积学习通道的尺寸规格修改为
1*1。
[0013]优选的,将
ECA
注意力机制模块融入
YOLOv5s
模型的
Backbone
的步骤如下:
[0014]步骤
A1
:将输入的特征图进过全局平均池化操作,将特征图的维度从
[h,w,c]缩减为
[1,1,c]的向量;
[0015]步骤
A2
:根据特征图的通道数,获取以为卷积核的大小;
[0016]步骤
A3
:根据卷积核的大小进行一维卷积操作,得到每个通道的权重;
[0017]步骤
A4
:将每一个通道的权重与原输入特征图进行逐通道相乘,得到加权后的特征图;
[0018]步骤
A5
:将
ECA
注意力模块融合到
YOLOv5s

Backbone
的四个
C3
模块中,形成
C3_ECA
模块

[0019]优选的,步骤
A2
中过去卷积核的大小公式如下:
[0020][0021]其中
c
为通道维数,
r

b
为超参数,其中
r
设置为
2、b
设置为1,
| |
odd
表示最接近的奇数

[0022]优选的,加权双向特征金字塔网络中进行加权特征融合的计算方式如下:
[0023][0024][0025]其中
P
iin
、P
itd
、P
iout
分别为第
i
层的输入特征

中间特征与输出特征,
Re size()
表示上采样或者下采样操作,
ε
表示是一个小的常数,用于防止分母为零,
w
k
、w

k
表示学习的权重参数

[0026]一种复杂环境下无人艇目标检测系统,使用所述一种复杂环境下无人艇目标检测方法,包括收集模块

改进模块

训练模块以及应用模块;
[0027]所述收集模块用于使用
LabelImg
对一定数量待检测物体图像进行标注,获得训练的数据集;
[0028]所述改进模块用于对
YOLOv5s
模型进行改进,其中改进包括将
ECA
注意力机制模块融入
YOLOv5s
模型的
Backbone
以及将
YOLOv5s
模型中将自底向上特征金字塔改为加权双向特征金字塔网络;
[0029]所述训练模块用于将数据集输入至改进后的
YOLOv5s
模型,对
YOLOv5s
模型进行训练;
[0030]所述应用模块用于将训练好的
YOLOv5s
模型嵌入至无人艇中,实现目标识别

[0031]上述技术方案中的一个技术方案具有如下优点或有益效果:在
YOLOv5
骨干本文档来自技高网
...

【技术保护点】

【技术特征摘要】
1.
一种复杂环境下无人艇目标检测方法,其特征在于,包括以下步骤:使用
LabelImg
对一定数量待检测物体图像进行标注,获得训练的数据集;对
YOLOv5s
模型进行改进,其中改进包括将
ECA
注意力机制模块融入
YOLOv5s
模型的
Backbone
以及将
YOLOv5s
模型中将自底向上特征金字塔改为加权双向特征金字塔网络;将数据集输入至改进后的
YOLOv5s
模型,对
YOLOv5s
模型进行训练;将训练好的
YOLOv5s
模型嵌入至无人艇中,实现目标识别
。2.
根据权利要求1所述一种复杂环境下无人艇目标检测方法,其特征在于,所述将
ECA
注意力机制模块融入
YOLOv5s
模型的
Backbone
前还需要执行如下操作:
ECA
注意力机制模块在
SE
模块上使用全连接层
FC
学习通道注意信息,并使用卷积学习通道的尺寸规格修改为
1*1。3.
根据权利要求2所述一种复杂环境下无人艇目标检测方法,其特征在于,将
ECA
注意力机制模块融入
YOLOv5s
模型的
Backbone
的步骤如下:步骤
A1
:将输入的特征图进过全局平均池化操作,将特征图的维度从
[h,w,c]
缩减为
[1,1,c]
的向量;步骤
A2
:根据特征图的通道数,获取以为卷积核的大小;步骤
A3
:根据卷积核的大小进行一维卷积操作,得到每个通道的权重;步骤
A4
:将每一个通道的权重与原输入特征图进行逐通道相乘,得到加权后的特征图;步骤
A5
:将
ECA
注意力模块融合到
YOLOv5s

Backbone
的四个
C3
模块中,形成

【专利技术属性】
技术研发人员:饶红霞曾鑫煌杨立鑫徐雍刘畅陶杰
申请(专利权)人:广东工业大学
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1