[发明专利]一种基于机器学习的柑橘果蒂的检测方法

摘要

本发明提供了一种基于机器学习的柑橘果蒂的检测方法，属于农业信息工程技术领域。本发明采用深度学习方法进行果蒂检测，构建网络模型以及检测流程，检测出果蒂的精准位置，虽然训练过程需要采集大量样本并花费比较长的时间进行训练，但是当训练完成后实际的检测过程计算量比较小，检测效率和准确率都适合在线系统的需求。本发明基于图像和机器学习技术提出了自动检测柑橘果蒂的方法，解决了柑橘自动整齐装箱和贴标签的关键技术。经过原型实现和仿真测试，证明本发明设计的方法实际可行且发挥了果蒂检测的功能。

主权项

1.一种基于机器学习的柑橘果蒂的检测方法，其特征在于，包括以下步骤：/n步骤1：图像的采集和预处理：使用数码相机采集柑橘照片，所述柑橘照片部分包含果蒂，剩余部分不包含果蒂，包含果蒂的用于产生正样本，不包含果蒂的用于产生负样本；/n步骤2：构建包含正负样本的训练集和测试集：将所述正样本和负样本分割成训练集和测试集；/n步骤3：计算每一张所述柑橘照片的待定区域，所述待定区域是根据亮度、颜色和纹理证据初步判断该位置附近可能存在果蒂的可疑对象，包括训练集中的待定区域和测试集中的待定区域；对于所述训练集中的每一张照片，计算各照片的亮度、颜色和纹理，然后将这些属性相似的邻近小区域合并，并反复迭代使得区域面积逐渐增大而总数逐渐减少，当所得整幅图像中剩余的区域减少到限定阈值时记录它们的外接矩形的坐标和长宽，以这些矩形作为所述待定区域；/n步骤4：构建网络模型：所述网络模型包括4个隐含层，第1至第3个隐含层为卷积层，用于提取图像的特征，第4个隐含层为全连接层，所述卷积层和全连接层的激活函数为ReLU函数；/n步骤5：训练网络模型的参数：将所述步骤3所得训练集中的待定区域输入到所述网络模型，以损失函数取得最小值为准则，使用反向传播方法自动调整网络参数，所述网络模型的输出包括两部分：第一部分是检测出的对象的类别，包括果蒂、果皮和其它，使用支持向量机进行分类，损失函数采用SoftMax函数；第二部分是外接矩形的横纵坐标、长度和宽度，使用线性回归模型，损失函数采用均方误差函数，训练完成后，将所得网络模型的结构和参数保存到h5文件中；/n步骤6：检测图像中是否存在果蒂并判断果蒂的位置：将所述步骤3所得测试集中的待定区域输入到所述步骤5中训练好的网络模型，判断是否包含果蒂并计算果蒂的位置。/n