基于旋转目标检测的稻穗谷粒计数与水稻产量估测方法、设备及介质

本发明公开了基于旋转目标检测的稻穗谷粒计数与水稻产量估测方法、设备及介质，其方法包括步骤：S1、获取稻穗数据集：构建水稻稻穗图像和对应产量的数据库并预处理；S2、对YOLOv8n_OBB模型轻量化改进得到YOLO‑Grain_OBB模型；S3、利用稻穗数据集训练YOLO‑Grain_OBB模型得到稻穗谷粒检测模型；S4、通过稻穗谷粒检测模型进行谷粒的识别与计数，输出谷粒的旋转目标检测框，并根据旋转目标检测框信息提取优选的谷粒表型特征；S5、选取不同的机器学习模型，对不同的优选的谷粒表型特征组合与产量数据进行拟合，构建筛选得到最优的产量预测模型。本发明的方法能够快速地对谷粒进行识别计数，并实现通过谷粒表型特征进行水稻产量的准确预测。