一种岩性场景分类模型构建方法及分类方法技术

技术编号:40638175 阅读:15 留言:0更新日期:2024-03-13 21:21
本发明专利技术提供一种岩性场景分类模型构建方法及分类方法,涉及图像处理技术领域,岩性场景分类模型构建方法包括获取地质遥感影像,构建数据集;根据数据集,训练初始分类模型,得到岩性场景分类模型,其中,初始分类模型包括场景分类子模型、语义分割子模型以及结果融合模块,场景分类子模型用于根据地质遥感影像输出第一分类图,语义分割子模型用于根据地质遥感影像输出第二分类图,结果融合模型用于融合第一分类图和第二分类图,生成岩性场景分类图;场景分类子模型包括特征提取模块、通道特征增强模块、上下文特征增强模块以及输出模块。通过数据集对初始分类模型不断训练,得到的岩性场景分类模型可提高岩性场景分类的精度。

【技术实现步骤摘要】

本专利技术涉及图像处理,具体而言,涉及一种岩性场景分类模型构建方法及分类方法


技术介绍

1、岩性场景分类图通常指的是地质学中的岩石类型和分布情况,岩性场景分类图包括不同类型的岩石,例如火成岩、沉积岩、变质岩等,以及它们在地质时间尺度上的分布和相互关系,岩性场景分类图对于研究地质历史、矿产资源勘探和环境地质等方面都具有重要意义。

2、在现有技术中,一般采用人工实地考察的方式来构建岩性场景分类图,但是,人工实地考察成本高,还存在部分人工不能涉及的区域,且由于在岩性场景的特征中,相似特征过多,导致构建的岩性场景分类精度低,不能反馈真实的岩性场景。


技术实现思路

1、本专利技术解决的问题是如何提高岩性场景的分类精度。

2、为解决上述问题,第一方面,本专利技术提供一种岩性场景分类模型构建方法,包括:

3、获取地质遥感影像,构建数据集;

4、根据所述数据集,训练初始分类模型,得到岩性场景分类模型,其中,所述初始分类模型包括场景分类子模型、语义分割子模型以及结果融合模块,所本文档来自技高网...

【技术保护点】

1.一种岩性场景分类模型构建方法,其特征在于,包括:

2.根据权利要求1所述的岩性场景分类模型构建方法,其特征在于,所述通道特征增强模块包括至少一个视觉通道注意力单元和第一特征融合单元,每个所述视觉通道注意力单元的输入端分别与所述特征提取模块的输出端连接,每个所述视觉通道注意力单元的输出端分别与所述第一特征融合单元的输入端连接;

3.根据权利要求2所述的岩性场景分类模型构建方法,其特征在于,所述视觉通道注意力单元包括第一提取子单元、第二提取子单元以及融合输出子单元,所述特征提取模块的输出端分别与所述第一提取子单元、所述第二提取子单元以及所述融合输出子单元的输入端连...

【技术特征摘要】

1.一种岩性场景分类模型构建方法,其特征在于,包括:

2.根据权利要求1所述的岩性场景分类模型构建方法,其特征在于,所述通道特征增强模块包括至少一个视觉通道注意力单元和第一特征融合单元,每个所述视觉通道注意力单元的输入端分别与所述特征提取模块的输出端连接,每个所述视觉通道注意力单元的输出端分别与所述第一特征融合单元的输入端连接;

3.根据权利要求2所述的岩性场景分类模型构建方法,其特征在于,所述视觉通道注意力单元包括第一提取子单元、第二提取子单元以及融合输出子单元,所述特征提取模块的输出端分别与所述第一提取子单元、所述第二提取子单元以及所述融合输出子单元的输入端连接,所述第一提取子单元与所述第二提取子单元的输出端分别与所述融合输出子单元的输入端连接,所述融合输出子单元的输出端与所述第一特征融合单元的输入端连接;

4.根据权利要求2所述的岩性场景分类模型构建方法,其特征在于,所述第一特征融合单元包括全局池化层、滤波层、第五全连接层、第二点乘层以及第二点加层,所述全局池化层、所述滤波层、所述第五全连接层、所述第二点乘层以及所述第二点加层顺次连接,且所述第二点乘层的输入端还接收所述通道注意力特征。

5.根据权利要求1所述的岩性场景分类模型构建方法,其特征在于,所述上下文特征增强模块包...

【专利技术属性】
技术研发人员:李显巨隋志远范小洁邹崇尧陈刚李义龙
申请(专利权)人:中国地质大学武汉
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1