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

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

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


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

技术介绍

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

技术实现思路

[0004]本专利技术的目的是提供一种基于深度学习的土豆图像识别系统，以解决上述现有技术存在的问题。
[0005]为实现上述目的，本专利技术提供了一种基于深度学习的土豆图像识别系统，包括：
[0006]图像获取模块、模型构建模块、图像识别模块和图像分类模块，其中所述图像获取模块、所述模型构建模块、所述图像识别模块和所述图像分类模块依次连接；
[0007]所述图像获取模块，用于获取不同环境生长的土豆本文档来自技高网...

【技术保护点】

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

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