摘要
本发明公开了一种茶叶采摘部位快速精准识别方法,首先获取茶叶图像,基于轻量化改进的目标检测模型生成茶叶嫩梢的检测框,通过轻量化改进的多任务模型同时进行茶叶嫩梢采摘关键点的定位及采摘区域的分割,最后提取采摘区域骨架,并进行融合关键点及细化骨架的茶叶嫩梢采摘点定位。本发明的方案基于YOLOv8网络模型进行轻量化改进,构建了目标检测模型以及多任务模型,自动学习茶叶关键点特征及嫩梢茎部特征,针对茶叶采摘这一需要精度的机器化操作,结合实例分割和关键点检测来定位茶叶嫩梢采摘点,从而保证了计算得出的采摘点一定位于茶叶嫩梢的茎部上,最大程度减少了定位误差,具有更好的泛化能力。
技术关键词
精准识别方法
实例分割
直方图均衡化方法
计算机可存储介质
亮度
标注工具
图片
残差模块
图像
关键点特征
掩膜
细化算法
通道
训练集
矫正
处理器
像素点
卷积模块