一种基于深度学习的土豆图像识别系统技术方案

本发明专利技术公开了一种基于深度学习的土豆图像识别系统，本发明专利技术属于图像识别领域，包括：图像获取模块、模型构建模块、图像识别模块和图像分类模块，通过图像获取模块，获取不同环境生长的土豆图像，将土豆图像分为训练图像、测试图像和待测图像；通过模型构建模块，构建神经网络识别模型，并基于训练图像训练神经网络识别模型，其中神经网络识别模型包括：卷积神经网络和全卷积神经网络，卷积神经网络用于提取土豆图像中的颜色特征，全卷积神经网络用于提取土豆图像中的尺寸特征；通过图像识别模块，将待测图像输入至神经网络识别模型中，得到土豆图像的识别结果；通过图像分类模块将待测图像进行分类。本发明专利技术能够提高土豆分类效率和分类准确度。和分类准确度。和分类准确度。

【技术实现步骤摘要】
一种基于深度学习的土豆图像识别系统


[0001]本专利技术属于图像识别领域，特别是涉及一种基于深度学习的土豆图像识别系统。

技术介绍

[0002]土豆作为重要的粮食来源，用途非常广泛。现在农业机械收割装置可以实现对土豆的大规模快速的收割，通过土豆分类装置对土豆进行分类和分拣，大多数基于土豆大小尺寸不同对土豆进行分类，将分类后的土豆进行不同价格销售。
[0003]现有技术中，通过土豆分类装置分拣土豆分类时间长，无法同时对土豆颜色、尺寸进行差异化分类，最终导致土豆分类不准确，影响土豆销售价格。

技术实现思路

[0004]本专利技术的目的是提供一种基于深度学习的土豆图像识别系统，以解决上述现有技术存在的问题。
[0005]为实现上述目的，本专利技术提供了一种基于深度学习的土豆图像识别系统，包括：
[0006]图像获取模块、模型构建模块、图像识别模块和图像分类模块，其中所述图像获取模块、所述模型构建模块、所述图像识别模块和所述图像分类模块依次连接；
[0007]所述图像获取模块，用于获取不同环境生长的土豆图像，将所述土豆图像分为训练图像、测试图像和待测图像；
[0008]所述模型构建模块，用于构建神经网络识别模型，并基于所述训练图像训练所述神经网络识别模型，其中所述神经网络识别模型包括：卷积神经网络和全卷积神经网络，所述卷积神经网络，用于提取所述土豆图像中的颜色特征，所述全卷积神经网络，用于提取所述土豆图像中的尺寸特征；
[0009]所述图像识别模块，用于将所述待测图像输入至所述神经网络识别模型中，得到土豆图像的识别结果；
[0010]所述图像分类模块，用于基于所述识别结果，将所述待测图像进行分类。
[0011]优选地，所述图像获取模块包括：图像获取单元和图像处理单元，其中所述图像获取单元、图像处理单元分别与所述模型构建模块连接；
[0012]所述图像获取单元，用于获取土豆图像，或者截取土豆视频中的土豆图像；
[0013]所述图像处理单元，用于将所述土豆图像转换成同样格式、同样分辨率，对所述土豆图像进行对比度增强的预处理操作；同时通过数据增强，随机水平翻转和随机打乱所述训练图像。
[0014]优选地，所述卷积神经网络采用AlexNet网络模型，所述AlexNet网络模型包括：依次连接的五个卷积层和三个全连接层，AlexNet网络模型采用ReLU非线性激活函数，AlexNet网络模型用于提取所述土豆图像中的颜色特征。
[0015]优选地，所述模型构建模块包括：模型构建单元、模型训练单元和模型测试单元；其中所述模型构建单元、所述模型训练单元和所述模型测试单元分别与所述图像识别模块
连接；
[0016]所述模型构建单元，用于构建AlexNet神经网络模型和全卷积神经网络；
[0017]所述模型训练单元，用于通过处理后的训练图像分别训练所述AlexNet神经网络模型、全卷积神经网络，得到训练好的AlexNet神经网络模型、全卷积神经网络；
[0018]所述模型测试单元，用于根据所述测试图像分别测试训练好的AlexNet神经网络模型、全卷积神经网络。
[0019]优选地，所述模型测试单元中，还包括计算测试误差，基于所述测试误差，分别调整AlexNet神经网络模型和全卷积神经网络的模型参数，基于调整后的模型参数，继续根据测试图像进行模型参数调整的过程，直至迭代次数达到预设次数。
[0020]优选地，所述图像识别模块包括：颜色分类单元、尺寸分类单元和特征识别单元，其中所述颜色分类单元、所述尺寸分类单元和所述特征识别单元分别与所述图像分类模块连接；
[0021]所述颜色分类单元，用于通过数据增强处理所述待测图像，并将处理后的待测图像输入至所述AlexNet神经网络模型中，输出分类结果；
[0022]所述尺寸分类单元，用于通过数据增强处理所述待测图像，并将处理后的待测图像输入至全卷积神经网络中，得到分类特征图，基于所述分类特征图，得到特征区域数据；
[0023]所述特征识别单元，用于重新构建特征识别模型，将所述分类结果和所述分类特征图同时输入至所述特征识别模型中，得到土豆图像的识别结果，所述识别结果为：黄色大土豆、黄色中土豆、黄色小土豆、白色大土豆、白色中土豆和白色小土豆。
[0024]优选地，所述尺寸分类单元中，基于所述分类特征图的图像轮廓，得到土豆区域，基于所述土豆区域计算特征区域数据，将所述特征区域数据与阈值比较，得到尺寸分类结果。
[0025]优选地，所述图像分类模块包括：标签标记单元和图像分类单元；
[0026]所述标签标记单元，用于基于识别结果，标记所述待测图像；
[0027]所述图像分类单元，用于将标记好的待测图像进行分类，同时将对应待测图像中的土豆进行分类。
[0028]本专利技术的技术效果为：
[0029]本专利技术通过图像获取模块，获取不同环境生长的土豆图像，将土豆图像分为训练图像、测试图像和待测图像；通过模型构建模块，构建神经网络识别模型，并基于训练图像训练神经网络识别模型，其中神经网络识别模型包括：卷积神经网络和全卷积神经网络，卷积神经网络，用于提取土豆图像中的颜色特征，全卷积神经网络，用于提取土豆图像中的尺寸特征；通过图像识别模块，将待测图像输入至神经网络识别模型中，得到土豆图像的识别结果；通过图像分类模块，将待测图像进行分类。本专利技术能够解决现有技术中土豆分类时间长的问题，通过网络模型进行图像识别，能够提高土豆分类效率，还可以同时对土豆颜色、尺寸进行差异化分类，提高土豆分类的准确度，最终能够获得土豆更加合理的销售价格。
附图说明
[0030]构成本申请的一部分的附图用来提供对本申请的进一步理解，本申请的示意性实施例及其说明用于解释本申请，并不构成对本申请的不当限定。在附图中：
[0031]图1为本专利技术实施例中的系统示意图；
[0032]图2为本专利技术实施例中的图像识别模块中的单元示意图。
具体实施方式
[0033]需要说明的是，在不冲突的情况下，本申请中的实施例及实施例中的特征可以相互组合。下面将参考附图并结合实施例来详细说明本申请。
[0034]实施例一
[0035]如图1所示，本实施例中提供一种基于深度学习的土豆图像识别系统，包括：
[0036]图像获取模块、模型构建模块、图像识别模块和图像分类模块，其中图像获取模块、模型构建模块、图像识别模块和图像分类模块依次连接；
[0037]图像获取模块，用于获取不同环境生长的土豆图像，将土豆图像分为训练图像、测试图像和待测图像；
[0038]模型构建模块，用于构建神经网络识别模型，并基于训练图像训练神经网络识别模型，其中神经网络识别模型包括：卷积神经网络和全卷积神经网络，卷积神经网络，用于提取土豆图像中的颜色特征，全卷积神经网络，用于提取土豆图像中的尺寸特征；
[0039]图像识别模块，用于将待测图像输本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于深度学习的土豆图像识别系统，其特征在于，包括：图像获取模块、模型构建模块、图像识别模块和图像分类模块，其中所述图像获取模块、所述模型构建模块、所述图像识别模块和所述图像分类模块依次连接；所述图像获取模块，用于获取不同环境生长的土豆图像，将所述土豆图像分为训练图像、测试图像和待测图像；所述模型构建模块，用于构建神经网络识别模型，并基于所述训练图像训练所述神经网络识别模型，其中所述神经网络识别模型包括：卷积神经网络和全卷积神经网络，所述卷积神经网络，用于提取所述土豆图像中的颜色特征，所述全卷积神经网络，用于提取所述土豆图像中的尺寸特征；所述图像识别模块，用于将所述待测图像输入至所述神经网络识别模型中，得到土豆图像的识别结果；所述图像分类模块，用于基于所述识别结果，将所述待测图像进行分类。2.根据权利要求1所述的基于深度学习的土豆图像识别系统，其特征在于，所述图像获取模块包括：图像获取单元和图像处理单元，其中所述图像获取单元、图像处理单元分别与所述模型构建模块连接；所述图像获取单元，用于获取土豆图像，或者截取土豆视频中的土豆图像；所述图像处理单元，用于将所述土豆图像转换成同样格式、同样分辨率，对所述土豆图像进行对比度增强的预处理操作；同时通过数据增强，随机水平翻转和随机打乱所述训练图像。3.根据权利要求1所述的基于深度学习的土豆图像识别系统，其特征在于，所述卷积神经网络采用AlexNet网络模型，所述AlexNet网络模型包括：依次连接的五个卷积层和三个全连接层，AlexNet网络模型采用ReLU非线性激活函数，AlexNet网络模型用于提取所述土豆图像中的颜色特征。4.根据权利要求3所述的基于深度学习的土豆图像识别系统，其特征在于，所述模型构建模块包括：模型构建单元、模型训练单元和模型测试单元；其中所述模型构建单元、所述模型训练单元和所述模型测试单元分别与所述图像识别模块连接；所述模型构建单元，用于构建AlexNet神经网络模型和全卷积神经网络；所述模型训练单元，用于通过...

【专利技术属性】
技术研发人员：马生旺，
申请(专利权)人：青海卓旺智慧信息科技有限公司，
类型：发明
国别省市：