公开(公告)号：CN118552786A

公开(公告)日：2024-08-27

申请号：CN202410703996.6

申请日：2024-06-03

Applicant: 中国地质大学(武汉)

Inventor： 李显巨 ,  胡翰 ,  刘祯慧 ,  邹崇尧 ,  范小洁 ,  常屹冉 ,  丁慧君 ,  陈刚

IPC: G06V10/764 ,  G06V10/774 ,  G06V10/26 ,  G06V20/10 ,  G06V10/82 ,  G06N3/09

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