一种基于改进YOLOv8-pose的成熟草莓采摘点定位方法及系统
申请号:CN202510902168
申请日期:2025-07-01
公开号:CN120783014A
公开日期:2025-10-14
类型:发明专利
摘要
本发明涉及一种基于改进YOLOv8‑pose的成熟草莓采摘点定位方法及系统,包括:采集棚内草莓的图像数据样本;基于改进的YOLOv8‑pose模型对所述待检测草莓图像进行成熟草莓采摘点定位,获取成熟草莓采摘点定位结果;其中,所述改进的YOLOv8‑pose模型通过训练集和验证集完成训练,所述训练集和验证集数据为标注成熟、草莓顶部关键点、草莓采摘关键点的草莓图像集。改进的YOLOv8‑pose模型中C2PSA模块,其金字塔分层注意力机制模块PSA,通过多头注意力机制和前馈神经网络,模型能够捕获到草莓更丰富的多尺度特征信息;同时模型中空间金字塔池化模块使用ReLU激活函数,模型训练更加稳定和高效。本发明能够实现在实际场景下提高目标检测模型定位成熟草莓采摘点的准确率。
技术关键词
采摘点定位方法
空间金字塔池化
检测草莓
金字塔分层
CSP结构
关键点
图像
训练集数据
模块
点定位系统
多尺度特征提取
多头注意力机制
输出特征
前馈神经网络
全局平均池化