摘要
本发明公开了一种基于WGB‑YOLO的目标采摘行中多类火龙果检测方法,包括如下步骤:步骤一、采用深度相机拍摄火龙果RGB图像与深度图像;步骤二、标注图像;步骤三、采用数据增强扩充标注的火龙果数据集,并将其划分出训练集与验证集;步骤四、构建WGB‑YOLO网络模型;步骤五、采用训练集训练WGB‑YOLO网络模型,获得WGB‑YOLO火龙果检测模型;步骤六、采用验证集验证WGB‑YOLO火龙果检测模型的检测精度;步骤七、将符合检测精度要求的WGB‑YOLO火龙果检测模型部署到密植果园果实采摘机器人移动终端系统上,用于检测目标采摘行的多类别火龙果。本发明提高了火龙果多类检测识别方法检测目标采摘行果实的精度,可以区分果实是否为目标采摘行的果实。