一种基于VIT与空间注意力机制的小麦品种鉴别方法

本发明公开了一种基于VIT与空间注意力机制的小麦品种鉴别方法，包括下述步骤：S1、图像采集：采用摄像设备进行小麦冠层图像和侧面株型的图像采集；S2、小麦品种数据集的构建：构建WMVD数据集，通过在原始图像上设定固定大小的窗口，逐步滑动以生成多个子图像；S3、小麦品种鉴选模型框架设计及特征提取网络的构建；S4、基于T‑SNE的高维特征可视化：对ViT+SPA模型提取的高维特征降维并实现特征的可视化；S5、基于小麦特征图的纹理性状提取。本发明实现增强了对小麦冠层图像和侧面株型图像的特征提取能力，构建了ViT+SPA模型以关注图像中的重要区域，提高了深度学习模型对不同小麦品种的识别精度。