基于P-MobileViT网络的小麦病害分类研究

打开文本图片集

doi：10.15889/j.issn.1002-1302.2024.20.031

摘要：针对小麦病害图像分类方法的识别准确率不理想、模型参数量大等问题，提出一种基于P-MobileViT的小麦病害分类模型。首先对小麦图像进行健康和病害二分类，融合Grabcut算法、大律法对小麦病害图像的病斑区域进行分割；然后将病斑图像输入P-MobileViT分类模型，在其block的局部表征模块中引入深度卷积提取病斑图像的局部特征，在全局表征模块使用PoolFormer结构提取全局特征，以减少模型计算量和参数量；将输入特征图和全局特征叠加后与局部特征进行融合，从而强化模型对特征的分类能力。（剩余11783字）