基于SK-EfficientNet的番茄叶片病害识别模型

摘要：

针对目前番茄叶片病害识别的深度学习模型网络参数量多、精确度低、移动端模型部署难的问题,提出一种基于SK-EfficientNet的番茄叶片病害识别方法.该方法采用轻量级模型EfficientNet作为基准模型,并利用选择性卷积核机制SKNet替换EfficientNet核心模块MBConv中的SENet,使得卷积核根据输入特征的多尺度信息自适应选择感受野大小,提高图像特征提取能力同时更有效地利用参数.多组对比实验结果显示,改进后的模型在训练精度上得到进一步提高,且模型参数仅为3.83 MiB.在PlantVillage数据集上平均准确率达到99.64％,且验证SK-EfficientNet-B2的识别精度最高;在自然场景下平均准确率较原模型提高3.81个百分点.结果表明,改进后模型能有效提高自然场景下番茄叶片病害识别精度,可为移动端部署番茄叶片病害识别模型提供参考.

关键词：番茄叶片、病害识别、EfficientNet网络、SKNet、MBConv

所属期刊栏目：40

分类号：TP391.41(计算技术、计算机技术)

资助基金：湖北省技术创新专项重大项目;湖北省科技重大专项;武汉市科技计划应用基础前沿项目

在线出版日期：2022-08-11（万方平台首次上网日期，不代表论文的发表时间）

页数：共11页

页码：104-114