【技术实现步骤摘要】
一种复杂环境下无人艇目标检测系统及方法
[0001]本专利技术涉及陶瓷胚体制作
,特别是一种复杂环境下无人艇目标检测系统及方法
。
技术介绍
[0002]当前,随着科技与社会需求的快速发展,无人平台的发展也越来越快
。
基于水上作业的需求,无人艇也得到迅速发展并使用各个水上作业工作中,如水上巡航
、
清洁
、
救援等
。
在复杂的水上环境进行目标识别任务时,目前常用的目标检测算法是
YOLOv5。
[0003]YOLOv5
骨干网络使用的是基于
CSPDarknet53
的骨干网络结构,虽然这些网络在提取单一尺度的特征方面表现良好,但缺少明确的多尺度信息融合机制,这会导致网络对不同尺度目标的检测能力不够强大,缺乏多尺度信息融合的问题
。
同时,应用于无人艇目标检测时,在复杂场景中存在大量遮挡
、
尺度变化等因素,这会导致骨干网络的特征提取能力受限,影响目标检测的准确性和鲁棒性
。
[0004]YOLOv5
中的
PANet
结构旨在改善特征融合过程,将骨干网络不同层级的特征进行融合
。
它通过侧连接将浅层特征与深层特征相结合,实现了高分辨率特征和低分辨率特征的融合
。
这使得模型能够同时捕捉到深层特征的语义信息和浅层特征的细节信息
。
但
PAnet
需要进行跨阶段的 ...
【技术保护点】
【技术特征摘要】 【专利技术属性】
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
模块中,形成
技术研发人员:饶红霞,曾鑫煌,杨立鑫,徐雍,刘畅,陶杰,
申请(专利权)人:广东工业大学,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。