基于小波变换与视觉ViT的高光谱图像分类方法及设备

本发明涉及基于小波变换与视觉ViT的高光谱图像分类方法及设备，属于图像处理和计算机视觉技术领域。先将图像块小波变换生成多个小波子带，然后沿通道进行拼接生成新的3D表示，用3D卷积提取具有局部上下文信息的特征图，再利用视觉ViT编码器执行自注意力机制学习，学习到长程依赖，最后分类得分类图。本发明通过将小波变换与视觉ViT结合,可以在保障分类质量的同时,显著提高基于视觉ViT的高光谱图像分类模型的计算效率。