【技术实现步骤摘要】
一种结合检测网络和点回归的葡萄图像采摘点位置单阶段定位方法
[0001]本专利技术涉及农业采摘
,具体来说是一种结合检测网络和点回归的葡萄图像采摘点位置单阶段定位方法
。
技术介绍
[0002]目前,葡萄采摘主要依靠自动化低
、
时效性差
、
成本高的人工操作
。
随着机器视觉
、
人工智能和机器人技术在园艺中的应用,自动采摘已成为一种新兴趋势
。
葡萄采摘通常是通过切割茎来执行的,以避免软皮肤的损伤
。
在这个过程中,茎中切割位置的测量,称为采摘点位置,是自动选择中必不可少的关键步骤
。
[0003]采摘点位置通常采用颜色和形态学的阈值方法来检测水果,然后利用统计计算得到采摘点
。
使用
Otsu
阈值来分离水果和茎,然后使用
Harris
角在茎上获得
。Otsu
阈值和颜色直方图分离用于识别番茄,然后将形态学和几何计算结合起来定位采摘点
。
泛化和容易噪声的能力是水果检测阈值方法的弱点之一,在复杂环境中导致鲁棒性较差和精度较低
。
随着深度学习
(DL)
的快速发展,将基于区域的卷积神经网络
(MASK R
‑
CNN)、Swin Transformer
和
You Only Look Once(YOLO)
等检测网络应用于水果检 ...
【技术保护点】
【技术特征摘要】 【专利技术属性】
1.
一种结合检测网络和点回归的葡萄图像采摘点位置单阶段定位方法,其特征在于,包括以下步骤:
11)
葡萄图像的获取及预处理:获取园内葡萄图像并进行预处理;
12)
葡萄采摘点定位模型的构建:基于
YOLOv5s
与点回归相结合构建葡萄采摘点定位模型;
13)
葡萄采摘点定位模型的训练:将预处理后的葡萄图像输入葡萄采摘点定位模型进行训练;
14)
待定位葡萄图像的获取:获取待定位葡萄图像并进行预处理;
15)
葡萄图像采摘点位置的定位:将预处理后的待定位葡萄图像输入训练后的葡萄采摘点定位模型,获得葡萄图像采摘点位置
。2.
根据权利要求1所述的一种结合检测网络和点回归的葡萄图像采摘点位置单阶段定位方法,其特征在于,所述葡萄采摘点定位模型的构建包括以下步骤:
21)
设定葡萄采摘点定位模型包括分为主干模块
、
颈部模块和头部模块;
22)
设定主干模块为
CSPDarknet53
;
23)
设定颈部模块包括特征金字塔网络
FPN
和像素聚合网络
PAN
,
FPN
结合低特征图和高特征图来传达深度语义信息,
PAN
使特征图包含图像语义和位置信息;
24)
设定头部模块包括为三点头部结构,其中的非最大抑制用于消除冗余边界框并保留最相关的边界框进行目标检测,三点头部结构提供了三个不同的尺度来检测不同大小的对象,分别为
20
×
20、40
×
40
和
80
×
80
,具有
12
维向量的头部输出标签包括置信度
、
分类
、
边界框和三个点;附加点使检测器输出候选采摘点
。3.
根据权利要求1所述的一种结合检测网络和点回归的葡萄图像采摘点位置单阶段定位方法,其特征在于,所述葡萄采摘点定位模型的训练包括以下步骤:
技术研发人员:张寒苏,苌浩宇,张盛羽,陈旭,占子朝,李海东,芦奇宝,李士昌,沈瓒骏,侯芷萌,
申请(专利权)人:安徽大学,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。