一种基于神经网络和迁移学习的水果图像分类方法及装置制造方法及图纸

本发明专利技术涉及一种基于神经网络和迁移学习的水果图像分类方法及装置，该方法包括：将待分类的图像进行预处理，经预处理后再结合SMOTE算法进行数据增强；将数据增强后的图像的像素点进行BN批量归一化操作，使所述像素点符合正态分布；将所述正态分布的像素点输入基于神经网络和迁移学习的水果图像分类模型，输出所述图像的分类结果。该方法可解决对输入的任意尺寸图像进行分类，该方法依赖基于神经网络和迁移学习的水果图像分类模型，提高识别和分类效率，降低大量时间成本，可靠性高。

【技术实现步骤摘要】
一种基于神经网络和迁移学习的水果图像分类方法及装置
本专利技术涉及深度学习领域，特别涉及一种基于神经网络和迁移学习的水果图像分类方法及装置。
技术介绍
目前卷积神经网络被广泛用于图像分析的应用场景当中，通过给定数据集上训练模型来完成特定任务，比如分类、检测、识别等。深度学习是图像分析研究中的一个新的领域，其目的在于模拟人脑进行自主学习。深度学习是一种数据驱动型模型，能够模拟人脑视觉机理自动地学习到数据各个层次的抽象特征，从而更好地反映数据的本质特征。随着图像库的不断增大，复杂度不断增高，单机的提取RGB图像特征已经远远不能满足需求，使用数据并行处理无疑是一个好的解决方案。传统图像分类算法主要有两种，一种是采用sift算法提取特征、进行分类；另一种是，采用神经网络全连接方式和反向传播算法(BP)的方式，利用高斯分布或随机初始值的方法，通过迭代的方式计算当前网络的输出的得分值，然后根据当前预测标签和实际标签之间得分值的差去不断调整前面各层之间的参数，直到整个模型的参数权重w和偏置b收敛。传统的sift算法，误差较大，且需结合大量及繁琐的图像预处理。传统的BP算法，存在着梯度弥散和爆炸问题本文档来自技高网...

【技术保护点】
1.一种基于神经网络和迁移学习的水果图像分类方法，其特征在于，包括：将待分类的图像进行预处理，经预处理后再结合SMOTE算法进行数据增强；将数据增强后的图像的像素点进行BN批量归一化操作，使所述像素点符合正态分布；将所述正态分布的像素点输入基于神经网络和迁移学习的水果图像分类模型，输出所述图像的分类结果。

【技术特征摘要】
1.一种基于神经网络和迁移学习的水果图像分类方法，其特征在于，包括：将待分类的图像进行预处理，经预处理后再结合SMOTE算法进行数据增强；将数据增强后的图像的像素点进行BN批量归一化操作，使所述像素点符合正态分布；将所述正态分布的像素点输入基于神经网络和迁移学习的水果图像分类模型，输出所述图像的分类结果。2.如权利要求1所述的基于神经网络和迁移学习的水果图像分类方法，其特征在于，将待分类的图像进行预处理，包括：将待分类的图像处理尺寸为299*299像素的图像。3.如权利要求1所述的基于神经网络和迁移学习的水果图像分类方法，其特征在于，所述数据增强的方式包括以一种或多种：缩放：将图像尺寸缩放80～120％，数据增加至2倍；镜像翻转：将原始图像进行垂直和水平反转，将图像数据扩充2倍；高斯噪声：将原始图像中添加一些高斯噪声，将图像数据扩充2倍；颜色抖动：增强或减弱原始图像对比度，将图像数据扩充2倍；旋转图片：对原始图像进行旋转45度，将图像数据扩充2倍。4.如权利要求1所述的基于神经网络和迁移学习的水果图像分类方法，其特征在于，所述结合SMOTE算法进行数据增强，公式如下：式中：特征向量xi,i∈{1,......,T}，T个样本，i为样本编号，ζ为0～1之间的随机数。5.如权利要求1所述的基于神经网络和迁移学习的水果图像分类方法，其特征在于，所述基于神经网络和迁移学习的水果图像分类模型由下述步骤训练生成，包括：采集大量水果图像，进行预处理；经预处理后再结合SMOTE算法，进行数据增强；将数据增强后的图像的像素点进行BN批量归一化操作，使所述像素点符合正态分布；构建一个卷积神经网络模型；所述卷积神经网络模型包含一个含有CNN、分类神经网络、输出网络三部分的卷积；将inceptionV3模型在ImageNet数据集上训练的权重迁移至所述卷积神经网络模型上；将所述正态分布的像素点输入迁移后的所述卷积神经网络模型，进行一次前向传播，输出得出每一分类得分值，通过softmax函数，得出每一类的概率值；计算所述得分值与所述真实分...

【专利技术属性】
技术研发人员：王卫星，黄仲强，姜晟，赖俊桂，韩清春，
申请(专利权)人：华南农业大学，
类型：发明
国别省市：广东,44