-
公开(公告)号:CN118552786A
公开(公告)日:2024-08-27
申请号:CN202410703996.6
申请日:2024-06-03
Applicant: 中国地质大学(武汉)
IPC: G06V10/764 , G06V10/774 , G06V10/26 , G06V20/10 , G06V10/82 , G06N3/09
Abstract: 本发明提供了一种分类模型的训练方法、高光谱图像分类方法、装置及设备,涉及遥感图像处理技术领域,分类模型的训练方法包括对处理后的高光谱图像顺次应用空间自注意和光谱自注意,生成空间光谱特征数据令牌;对所述空间光谱特征数据令牌分别进行重构和对比多视图学习,生成重构损失和对比损失;基于所述重构损失和所述对比损失,训练分类模型,得到预训练模型;基于所述预训练模型,根据所述高光谱图像和对应标签,生成交叉熵损失,并根据所述交叉熵损失,微调所述预训练模型,得到最终分类模型。本发明不仅不需要大量的样本数据,还会更好的训练模型,极大地提高分类模型的精度。
-
公开(公告)号:CN117689960B
公开(公告)日:2024-04-26
申请号:CN202410134343.0
申请日:2024-01-31
Applicant: 中国地质大学(武汉)
IPC: G06V10/764 , G06V20/10 , G06V10/44 , G06V10/80 , G06V10/82 , G06N3/0464
Abstract: 本发明提供一种岩性场景分类模型构建方法及分类方法,涉及图像处理技术领域,岩性场景分类模型构建方法包括获取地质遥感影像,构建数据集;根据数据集,训练初始分类模型,得到岩性场景分类模型,其中,初始分类模型包括场景分类子模型、语义分割子模型以及结果融合模块,场景分类子模型用于根据地质遥感影像输出第一分类图,语义分割子模型用于根据地质遥感影像输出第二分类图,结果融合模块用于融合第一分类图和第二分类图,生成岩性场景分类图;场景分类子模型包括特征提取模块、通道特征增强模块、上下文特征增强模块以及输出模块。通过数据集对初始分类模型不断训练,得到的岩性场景分类模型可提高岩性场景分类的精度。
-
公开(公告)号:CN117689960A
公开(公告)日:2024-03-12
申请号:CN202410134343.0
申请日:2024-01-31
Applicant: 中国地质大学(武汉)
IPC: G06V10/764 , G06V20/10 , G06V10/44 , G06V10/80 , G06V10/82 , G06N3/0464
Abstract: 本发明提供一种岩性场景分类模型构建方法及分类方法,涉及图像处理技术领域,岩性场景分类模型构建方法包括获取地质遥感影像,构建数据集;根据数据集,训练初始分类模型,得到岩性场景分类模型,其中,初始分类模型包括场景分类子模型、语义分割子模型以及结果融合模块,场景分类子模型用于根据地质遥感影像输出第一分类图,语义分割子模型用于根据地质遥感影像输出第二分类图,结果融合模型用于融合第一分类图和第二分类图,生成岩性场景分类图;场景分类子模型包括特征提取模块、通道特征增强模块、上下文特征增强模块以及输出模块。通过数据集对初始分类模型不断训练,得到的岩性场景分类模型可提高岩性场景分类的精度。
-
-