[发明专利]遥感图片样本的生成方法、装置及电子设备

说明书

本申请实施例提供了一种遥感图片样本的生成方法、装置及电子设备，其中方法包括：建立多个预设遥感目标的立体模型，得到每个预设遥感目标对应的第一模型；基于初始背景图建立对应的立体模型，得到第二模型；根据多个子区域和预设的子区域与类型的对应关系，查询每个子区域对应的目标类型；根据每个子区域对应的目标类型和每个子区域的区域面积，将每个第一模型填充至第二模型中，以生成目标模型；基于目标模型，生成对应的遥感图片样本；遥感图片样本包括虚拟遥感图片以及虚拟遥感图片的样本标注信息。以此，以初始背景图为基础，生成包括多个预设遥感目标的虚拟遥感图片，该虚拟遥感图片携带有样本标注信息，进而提高了遥感图片样本的生成效率。

技术领域

本申请图像处理技术领域，尤其涉及一种遥感图片样本的生成方法、装置及电子设备。

背景技术

随着电子技术的发展，遥感技术的应用越来越广泛。遥感(remotesensing)是指非接触的，远距离的探测技术，通过遥感器这类对电磁波敏感的仪器，在远离目标和非接触目标物体条件下探测目标地物。

在涉及遥感技术的模型训练过程中，往往需要海量的携带有标注信息的遥感图片作为训练样本。然而，一方面，由卫星采集的真实遥感图片价格昂贵，获取难度高，另一方面，真实遥感图像中各个物体较小，通过肉眼识别物体边缘难度较高，通过人工对真实遥感图片进行样本标注的工作量巨大且准确性较低，使得训练样本的生成难度较高，且训练样本的生成效率低下。

发明内容

本申请实施例的目的是提供一种遥感图片样本的生成方法、装置、电子设备，以解决如何提高遥感图片样本的生成效率的问题。

为解决上述技术问题，本申请实施例是这样实现的：

第一方面，本申请实施例提供了一种遥感图片样本的生成方法，该方法包括：

建立多个预设遥感目标的立体模型，得到每个预设遥感目标对应的第一模型；多个预设遥感目标包括多个类型的遥感目标；

基于初始背景图建立对应的立体模型，得到第二模型；初始背景图为不包括预设遥感目标的真实遥感图片；第二模型的底面包括多个子区域；

根据多个子区域和预设的子区域与类型的对应关系，查询每个子区域对应的目标类型；每个子区域对应的目标类型为用于填充每个子区域的遥感目标的类型；

根据每个子区域对应的目标类型和每个子区域的区域面积，将每个第一模型填充至第二模型中，以生成目标模型；

基于目标模型，生成对应的遥感图片样本；遥感图片样本包括虚拟遥感图片以及虚拟遥感图片的样本标注信息。

第二方面，本申请实施例提供了一种图像检测模型的训练方法，该方法包括：

通过上述第一方面的遥感图片样本的生成方法生成遥感图片样本；

基于遥感图片样本，对待训练的图像检测模型进行训练；训练后的图像检测模型用于对待检测的遥感图片进行目标检测。

第三方面，本申请实施例提供了一种图像检测方法，该方法包括：

获取待检测的遥感图片；

将待检测的遥感图片输入训练后的图像检测模型进行目标检测，得到待检测的遥感图片的图像检测结果；训练后的图像检测模型是基于遥感图片样本对待训练的图像检测模型进行训练所得到的；遥感图片样本是通过上述第一方面的遥感图片样本的生成方法所生成的。

第四方面，本申请实施例提供了一种遥感图片样本的生成装置，包括：