(19)国家知识产权局 (12)发明 专利申请 (10)申请公布号 (43)申请公布日 (21)申请 号 202210945258.3 (22)申请日 2022.08.08 (71)申请人 北京理工大 学 地址 100081 北京市海淀区中关村南大街5 号 申请人 中国地质大 学 （北京） (72)发明人 贾丽娟　何锐曦　张金川　 (74)专利代理 机构 石家庄国为知识产权事务所 13120 专利代理师 刘少卿 (51)Int.Cl. G06V 10/26(2022.01) G06V 10/44(2022.01) G06V 10/82(2022.01) G06V 20/13(2022.01)G06N 3/04(2006.01) G06N 3/08(2006.01) (54)发明名称 地热有利区的辨识方法、 装置、 终端及存储 介质 (57)摘要 本发明提供一种地热有利区的辨识方法、 装 置、 终端及存储介质， 该方法包括： 获取待辨识区 域的地表温度反演图像； 基于经过训练的温度分 割模型对所述地表温度反演图像进行区域划分， 得到所述地表温度反演图像中的地热有利区； 所 述温度分割模型为加入注意力机制的U ‑Net分割 网络。 本发 明通过温度分割模型对地表温度反演 图像进行分割， 可以在图像中提取高温区域的清 晰轮廓， 从而准确辨识出待辨识区域中的地热有 利区。 权利要求书2页 说明书10页 附图4页 CN 115272679 A 2022.11.01 CN 115272679 A 1.一种地热有利区的辨识方法， 其特 征在于， 包括： 获取待辨识区域的地表温度反演图像； 基于经过训练 的温度分割模型对所述地表温度反演图像进行区域划分， 得到所述地表 温度反演图像中的地热有利区； 所述温度分割模型为加入注意力机制的U ‑Net分割网络 。 2.根据权利要求1所述的地热有利区的辨识方法， 其特征在于， 在所述基于经过训练 的 温度分割模型对所述地表温度反演图像进 行区域划分， 得到所述地表温度反演图像中的地 热有利区之前， 所述方法还 包括： 建立初始的温度分割模型； 基于训练样本集对所述初始的温度分割模型进行训练， 得到经过训练的温度分割模 型； 其中， 所述训练样本集包括多个训练样本， 训练样本为经过标注的地表温度反演图像， 所述训练样本集中的各个训练样本涵盖四季， 并且各地表温度反演图像的含云量均低于预 设比例。 3.根据权利要求2所述的地热有利区的辨识方法， 其特征在于， 在所述基于训练样本集 对所述初始的温度分割模型进行训练， 得到经过训练的温度分割模型之前， 所述方法还包 括： 获取多个地表温度反演图像； 基于预设温度阈值对所述多个地表温度反演图像进行二值化处理， 得到多个训练样 本， 并基于所述多个训练样本构建训练样本集。 4.根据权利要求1所述的地热有利区的辨识方法， 其特征在于， 所述温度分割 模型包括 四层深度学习 卷积对和注意力模块， 每层深度学习 卷积对包括一个编码器和 一个解码器； 其中， 四个编 码器按照层数由小到大依次串联， 四个解码 器按照层数由大到小依次串联， 每 层编码器的图像尺寸是上一层编码器的图像尺寸的一半， 每层编 码器的通道数量是上一层 编码器的通道数量的两倍， 每层解码器的图像尺寸是上一层解码器的图像尺寸的两倍， 每 层解码器的通道数量是 上一层解码器的通道数量的一半。 5.根据权利要求1所述的地热有利区的辨识方法， 其特征在于， 所述温度分割 模型的损 失函数为BC E‑Dice损失函数。 6.根据权利要求1所述的地热有利区的辨识方法， 其特征在于， 所述获取待辨识区域的 地表温度反演图像， 包括： 获取待辨识区域的遥感图像； 利用ENVI遥感图像处理平台对所述待辨识区域的遥感图像进行地表温度反演， 得到待 辨识区域的地表温度反演图像。 7.一种地热有利区的辨识装置， 其特 征在于， 包括： 获取模块， 用于获取待辨识区域的地表温度反演图像； 分割模块， 用于基于经过训练的温度分割模型对所述地表温度反演图像进行区域划 分， 得到所述地表温度反演图像中的地热有利区； 所述温度分割模型为加入注意力机制的 U‑Net分割网络 。 8.根据权利要求7 所述的地热有利区的辨识装置， 其特 征在于， 所述装置还 包括： 建模模块， 用于在所述基于经过训练 的温度分割 模型对所述地表温度反演图像进行区 域划分， 得到所述 地表温度反演图像中的地热有利区之前， 建立初始的温度分割模型；权　利　要　求　书 1/2 页 2 CN 115272679 A 2训练模块， 用于基于训练样本集对所述初始 的温度分割模型进行训练， 得到经过训练 的温度分割模型； 其中， 所述训练样本集包括多个训练样 本， 训练样本为经过标注的地表温 度反演图。 9.一种终端， 包括存储器、 处理器以及存储在所述存储器中并可在所述处理器上运行 的计算机程序， 其特征在于， 所述处理器执行所述计算机程序时实现如上的权利要求1至6 中任一项所述方法的步骤。 10.一种计算机可读存储介质， 所述计算机可读存储介质存储有计算机程序， 其特征在 于， 所述计算机程序被处 理器执行时实现如上的权利要求1至 6中任一项所述方法的步骤。权　利　要　求　书 2/2 页 3 CN 115272679 A 3