[发明专利]一种基于SK-EfficientNet的轻量级农作物病害识别方法

摘要

本发明公开了一种基于SK‑EfficientNet的轻量级农作物病害识别方法，包括以下步骤：训练阶段：获取植物病害图像作为输入训练数据集；对训练数据集中的植物病害图像进行预处理；构建基于SK‑EfficientNet的分类模型；针对EfficientNet的核心模块移动翻转瓶颈卷积MBConv进行改进，将原有SENet模块替换为SKNet，形成SK‑MBConv网络结构；将SK‑MBConv融合原始EfficientNet模型，形成SK‑EfficientNet网络模型；对SK‑EfficientNet分类模型进行学习训练，保存训练好的模型；测试阶段：输入待识别的植物图像，通过训练好的SK‑EfficientNet分类模型对植物图像进行识别。本发明结合EfficientNet高速度与精度、SKNet高特征提取能力的优点，在减少模型参数的同时更好地识别出不同农作物的多种病害类别，尤其是复杂的农作物病害，并具有更高的识别精度。