[发明专利]一种基于SqueezeNet的农作物叶片病害识别方法

说明书

本发明公开了一种基于SqueezeNet的农作物叶片病害识别方法。其方案是：对原始数据集进行增强和扩充,划分训练集和测试集；从网络规模小型化和计算过程轻量化的角度出发，对经典SqueezeNet结构进行精简与参数修改，获取4种改进后SqueezeNet模型；训练参数设置,多次迭代后得到训练后的模型；将测试集输入训练后的模型,对多种农作物的不同叶片病害进行分类。本发明提出的改进SqueezeNet模型在显著减少参数内存要求和计算量的同时使模型性能保持在一个较高的水平，较好地平衡了这三项指标，有利于将模型部署在移动终端等嵌入式资源受限设备上，有助于实现对农作物病害的实时准确识别。

技术领域

本发明涉及农业植保领域，尤其涉及一种基于SqueezeNet的农作物叶片病害识别方法。

背景技术

准确识别农作物病害类别是农作物病害防治的前提，而农作物叶片病斑及其相关特征是判断农作物病害种类及其病害程度的重要依据。传统的农作物病害检测主要依靠人工现场观察判断，存在主观性强、工作强度大等不足。

利用现代信息技术对农作物病害种类进行诊断和识别是一种先进有效的手段。传统机器学习病害识别方法一般包含图像分割，特征提取和模式识别三个环节，如果不能准确的提取病斑底层特征并选择对分类贡献率较高的特征来进行分类，则分类性能会明显下降。与传统机器学习方法相比，深度学习是一种端到端的方法，它以原始数据为输入，以最终的任务为输出，经过层层抽取将原始数据逐层抽象为任务自身所需要的特征，避免了人为特征选取对分类性能的影响，也同时明显增强系统的识别性能。

目前卷积神经网络在农业工程相关领域已获得广泛应用。为了获得更好的性能，近年来网络层数不断增加，从7层的AlexNet到16层的VGGNet，再到22层的GoogleNet、152层的ResNet，更有上千层的ResNet和DenseNet等。但这些传统卷积神经网络识别系统存在模型参数大、模型运算量要求高的不足，简言之就是效率问题。

效率问题主要是模型的存储问题和模型计算量问题。首先，深层次网络需要保存大量权值参数，这对设备内存的要求较高；其次，在实际应用中往往是快速响应需求，为达到实用标准，要么提高处理器性能，要么减少计算量。只有解决CNN效率问题，才能让CNN走出实验室，更广泛的应用于日常生活当中。对此，通常的方法是进行模型压缩，即在已经训练好的模型上进行压缩，使得网络携带更少的网络参数，从而解决内存问题，同时可以解决计算速度问题。

相比于在已经训练好的模型上进行处理，轻量化模型设计则是另辟蹊径。轻量化模型设计的主要思想在于设计更高效的“网络计算方式”从而使网络参数及计算量减少的同时不过多损失网络的识别性能。

发明内容

发明的目的：本发明克服了传统卷积神经网络识别系统存在的模型参数量大、模型运算量要求高的不足，在经典SqueezeNet模型的基础上进行改进，提出一种基于SqueezeNet的农作物叶片病害识别方法。

技术方案：

本发明公开了一种基于SqueezeNet的农作物叶片病害识别方法，包括如下连续步骤：

1)收集各类不同农作物的不同种类叶片病害图像，对原始数据集进行增强和扩充,划分训练集和测试集；

2)从网络规模小型化和计算过程轻量化的角度出发，对经典SqueezeNet结构进行精简与参数修改，获取4种改进SqueezeNet模型；

3)训练参数设置,多次迭代后得到训练后的模型；

4)将测试图像输入训练后的模型中进行测试。

2、根据权利要求1所述的一种基于SqueezeNet的农作物叶片病害识别方法，其特征在于，所述1)中的4种改进SqueezeNet模型的获取包括如下步骤：