本公开涉及图像处理，尤其涉及一种建筑物屋顶信息提取方法、装置、电子设备及存储介质。
背景技术：
1、随着传感器技术的不断进步，遥感设备的清晰度也在不断提高。遥感影像技术是一种利用遥感设备获取地表信息的影像技术，已经广泛应用于农业、城市规划、环境监测等领域。通过高分辨率的图像捕捉，遥感影像技术不仅能提供更精细的地表观察，还能通过多时相、多波段的数据分析，为决策者提供动态变化的地表信息。
技术实现思路
1、本公开提供了一种建筑物屋顶信息提取方法、装置、电子设备及存储介质。
2、根据本公开的第一方面，提供了一种建筑物屋顶信息提取方法，包括：
3、获取第一语义分割样本数据；第一语义分割样本数据包括第一影像切片数据和与第一影像切片数据对应的二值图像数据；第一影像切片数据是对第一原始遥感影像数据进行随机取样得到的图像切片数据；二值图像数据是对第一影像切片数据中的建筑物屋顶进行标注得到的；
4、基于第一语义分割样本数据对第一模型进行训练，得到语义分割模型；第一模型以遥感影像数据为输入，以语义分割图像数据为输出；
5、将第一原始遥感影像数据输入到语义分割模型，得到第一语义分割图像数据，并对第一语义分割图像数据进行后处理，得到多张单个屋顶图像数据；
6、基于第一屋顶图像数据集对第二模型进行训练，得到屋顶图像分类模型；第二模型以单个屋顶图像数据为输入，以屋顶多分类信息为输出；第一屋顶图像数据集包括多张单个屋顶图像数据中至少一张单个屋顶图像数据和其对应的建筑物屋顶多分类信息；建筑物屋顶多分类信息是对对应的单个屋顶图像数据的建筑物屋顶进行多分类标注得到的；
7、将多张单个屋顶图像数据输入到屋顶图像分类模型，得到第一原始遥感影像数据中每个建筑物屋顶的多分类信息。
8、根据本公开的第二方面，提供了一种建筑物屋顶信息提取装置，包括：
9、获取模块，用于获取第一语义分割样本数据；第一语义分割样本数据包括第一影像切片数据和与第一影像切片数据对应的二值图像数据；第一影像切片数据是对第一原始遥感影像数据进行随机取样得到的图像切片数据；二值图像数据是对第一影像切片数据中的建筑物屋顶进行标注得到的；
10、第一训练模块，用于基于第一语义分割样本数据对第一模型进行训练，得到语义分割模型；第一模型以遥感影像数据为输入，以语义分割图像数据为输出；
11、后处理模块，用于将第一原始遥感影像数据输入到语义分割模型，得到第一语义分割图像数据，并对第一语义分割图像数据进行后处理，得到多张单个屋顶图像数据；
12、第二训练模块，用于基于第一屋顶图像数据集对第二模型进行训练，得到屋顶图像分类模型；第二模型以单个屋顶图像数据为输入，以屋顶多分类信息为输出；第一屋顶图像数据集包括多张单个屋顶图像数据中至少一张单个屋顶图像数据和其对应的建筑物屋顶多分类信息；建筑物屋顶多分类信息是对对应的单个屋顶图像数据的建筑物屋顶进行多分类标注得到的；
13、分类模块，用于将多张单个屋顶图像数据输入到屋顶图像分类模型，得到第一原始遥感影像数据中每个建筑物屋顶的多分类信息。
14、根据本公开的第三方面，提供了一种电子设备，包括：
15、至少一个处理器；以及
16、与至少一个处理器通信连接的存储器；其中，
17、存储器存储有可被至少一个处理器执行的指令，指令被至少一个处理器执行，以使至少一个处理器能够执行上述第一方面的方法。
18、根据本公开的第四方面，提供了一种存储有计算机指令的非瞬时计算机可读存储介质，该计算机指令用于使计算机执行上述第一方面的方法。
19、根据本公开实施例的第五方面，提供一种计算机程序产品，包括计算机程序，计算机程序在被处理器执行时实现上述第一方面中任一项方法的步骤。
20、本公开的技术方案：获取第一语义分割样本数据；基于第一语义分割样本数据对第一模型进行训练，得到语义分割模型；将第一原始遥感影像数据输入到语义分割模型，得到第一语义分割图像数据，并对第一语义分割图像数据进行后处理，得到多张单个屋顶图像数据；基于第一屋顶图像数据集对第二模型进行训练，得到屋顶图像分类模型；将多张单个屋顶图像数据输入到屋顶图像分类模型，得到第一原始遥感影像数据中每个建筑物屋顶的多分类信息，这样可以降低建筑物屋顶信息提取任务的整体难度，降低遥感影像的成像质量影响，以便于提高建筑物屋顶信息提取的准确度。
21、应当理解，本部分所描述的内容并非旨在标识本公开的实施例的关键或重要特征，也不用于限制本公开的范围。本公开的其它特征将通过以下的说明书而变得容易理解。
1.一种建筑物屋顶信息提取方法，其特征在于，包括以下步骤：
2.根据权利要求1所述的方法，其特征在于，所述基于所述第一语义分割样本数据对第一模型进行训练，得到语义分割模型，包括：
3.根据权利要求2所述的方法，其特征在于，所述第一模型是在u-net图像分割网络的基础上增加边缘填充操作和批标准化层的图像分割模型。
4.根据权利要求1所述的方法，其特征在于，所述对所述第一语义分割图像数据进行后处理，得到多张单个屋顶图像数据，包括：
5.根据权利要求4所述的方法，其特征在于，所述对所述中间语义分割图像数据进行连通域分析，得到至少一个第一连通域影像数据，包括：
6.根据权利要求1所述的方法，其特征在于，所述基于第一屋顶图像数据集对第二模型进行训练，得到屋顶图像分类模型，包括：
7.根据权利要求1-6中任意一项所述的方法，其特征在于，所述第一原始遥感影像数据中每个所述建筑物屋顶的多分类信息包括以下至少一项：
8.一种建筑物屋顶信息提取装置，其特征在于，包括：
9.一种电子设备，其特征在于，包括：
10.一种存储有计算机指令的非瞬时计算机可读存储介质，其特征在于，所述计算机指令用于使所述计算机执行权利要求1-7中任一项所述的方法。
