融合迁移学习和知识蒸馏的轻量级马铃薯叶片病害识别模型的构建方法

扫码查看

原文链接

万方数据
维普

中文摘要：轻量级深度学习模型常被部署于移动端或物联网端,以实现算力资源受限条件下马铃薯病害的识别.但轻量级模型网络层数较少,模型特征提取能力有限,无法实现相似表型特征的精确提取.为解决上述问题,提出一种轻量级残差网络模型的构建方法,该方法融合迁移学习和知识蒸馏策略训练模型,在教师模型上使用迁移学习策略缩短教师模型的训练时间,并将ResNet18模型进行模型剪枝,使用降采样的方法提高模型识别准确率,最终在保证轻量化的前提下,实现对马铃薯叶片病害类别的精准识别.在马铃薯叶片数据集上进行试验,结果表明,本研究方法构建的轻量级模型的识别准确率相较于Resnet18提高1.55百分点,模型大小缩小49.18％;相较于目前农作物病害识别领域,常用的轻量级模型MobileNetV3在模型大小相近的情况下,识别准确率提高2.91百分点.该模型能够满足大部分实际应用下的场景,可为部署在物联网和移动端设备上的模型提供参考.

作者：

章广传、李彤、高泉、叶荣、何云

展开 >

作者单位：

云南农业大学机电工程学院,云南昆明 650000

云南省作物生产与智慧农业重点实验室,云南昆明 650000

云南农业大学大数据学院,云南昆明 650000

关键词：

马铃薯病害识别轻量级模型迁移学习知识蒸馏

基金：

国家自然科学基金云南省基础研究计划云南省科技重大专项云南省科技重大专项省部共建云南生物资源保护与利用国家重点实验室开放基金

项目编号：

32101611202101AU070096202202AE090021202302AE090020GZKF2021009

出版年：

2024

DOI：

10.15889/j.issn.1002-1302.2024.04.030

江苏农业科学

江苏省农业科学院

江苏农业科学

CSTPCD北大核心

影响因子：0.732

ISSN：1002-1302

年,卷(期)：2024.52(4)

参考文献量23