-
公开(公告)号:CN117671370A
公开(公告)日:2024-03-08
申请号:CN202311671795.4
申请日:2023-12-07
Applicant: 北京林业大学
IPC: G06V10/764 , G06V10/77 , G06V10/40 , G06V10/82 , G06N3/0464 , G06N3/096
Abstract: 一种基于注意力和解耦合知识蒸馏的无患子病害分类方法。本发明公开了基于注意力和解耦合知识蒸馏的无患子病害分类方法,属于计算机视觉技术领域,包括以下步骤:(1)通过在无患子种植基地实地拍照的方式建立无患子病害数据集;(2)修改ResNet101结构并添加两种注意力模块提高模型表征能力,将改进的模型作为教师模型并训练;(3)基于解耦知识蒸馏实现无患子病害分类模型的压缩,将大模型轻量化,便于移植到移动端设备。本发明应用于无患子病害分类分类中,通过添加注意力提高教师模型表征能力;基于解耦知识蒸馏实现了无患子病害分类模型的高效压缩,提升学生模型的预测精度,最终实现高准确率的轻量级无患子病害分类模型。