摘要
本发明属于目标检测领域,目的是提供一种基于改进型YOLOv5的猕猴桃授粉花朵识别方法,以解决由于猕猴桃花朵重叠与具有一定倾斜角导致难以识别,实现充分授粉的问题。技术方案是一种改进型YOLOv5的猕猴桃授粉花朵识别方法,具体包括以下步骤:步骤S1:采集猕猴桃花朵图片;步骤S2:对采集到的图像数据进行数据增强;步骤S3:对数据集中的猕猴桃花朵进行标注,再分为训练集、验证集和测试集;步骤S4:在YOLOv5模型中的K‑means聚类数据算法更改成K‑means++算法;步骤S5:在YOLOv5模型中添加了CBMA注意力机制;步骤S6:将YOLOv5中GIOU‑loss损失函数更改成CIOU‑loss损失函数;步骤S7:在detect函数中加入花朵角度计算模块;步骤S8:在detect函数中加入重叠花朵情况判断及授粉点寻找模块;步骤S9:训练和测试。