基于轻量级网络自适应特征提取的番茄病害识别

Tomato disease recognition based on lightweight network auto-adaptive feature extraction

为了实现番茄病害的精准识别,本研究提出一种轻量级网络自适应特征提取方法。该方法首先对图片进行正形处理,然后基于SqueezeNet模型构建轻量级网络模型GKFENet。GKFENet模型包含全局特征提取和关键特征提取2个模块,其中全局特征提取模块逐层提取番茄病害叶片的全局特征,关键特征提取模块通过学习评估出特征图各通道的重要程度,计算出权重值,最后将该值加权到原特征图上,从而实现病害关键特征的自适应提取。结果显示,正形机制有助于神经网络学习特征,本研究构建的GKFENet模型的平均识别准确率为97.90%,模型大小仅为2.64 MB,且在强噪声环境下,其识别准确率仍能保持在78.00%以上。GKFENet模型在训练过程中相对稳定,对8种番茄病害的识别准确率均超过96.00%。相比Bayes、KNN、LeNet、SqueezeNet、MobileNet模型,本研究构建的GKFENet模型的识别精度高,稳定性强且占用内存小,对于移动端未来的应用具有较高的实际价值。

To realize accurate recognition of tomato diseases,a lightweight network auto-adaptive feature extraction method was proposed.This method firstly performed a correction processing on the image.Then,based on the SqueezeNet model,a lightweight network model named global and key feature extraction network(GKFENet)was built.The GKFENet model included global feature extraction and key feature extraction modules.The global feature extraction module extracted the global features of tomato diseased leaves layer by layer.The key feature extraction module evaluated the importance of each channel in the feature images through learning,and calculated the weight value.Finally,the value was weighted on the original feature images to realize the adaptive extraction of the key features of the diseases.The results showed that the correction mechanism helped the neural network to learn features,the average identification accuracy of GKFENet model constructed in this research was 97.90%,and the model size was only 2.64 MB.In an environment with strong noise,the recognition accuracy of the model was still above 78.00%.GKFENet model was relatively stable during the training process,and the recognition accuracies of eight types of tomato diseases were all over 96.00%.Compared with Bayes,k-nearest neighbor(KNN),LeNet,SqueezeNet and MobileNet models,the GKFENet model constructed in this research has higher recognition accuracy,stronger stability and less memory.It has strong practical value in future mobile applications.