摘要
本发明提供了一种分类模型的训练方法、高光谱图像分类方法、装置及设备,涉及遥感图像处理技术领域,分类模型的训练方法包括对处理后的高光谱图像顺次应用空间自注意和光谱自注意,生成空间光谱特征数据令牌;对所述空间光谱特征数据令牌分别进行重构和对比多视图学习,生成重构损失和对比损失;基于所述重构损失和所述对比损失,训练分类模型,得到预训练模型;基于所述预训练模型,根据所述高光谱图像和对应标签,生成交叉熵损失,并根据所述交叉熵损失,微调所述预训练模型,得到最终分类模型。本发明不仅不需要大量的样本数据,还会更好的训练模型,极大地提高分类模型的精度。
技术关键词
空间光谱特征
预训练模型
令牌
训练分类模型
数据
光谱图像分类方法
遥感图像处理技术
生成重构图像
编码器
索引
样本
存储计算机程序
多层感知机
标签
编码块