场景分类模型的训练方法、遥感影像场景分类方法及设备与流程

本发明涉及人工智能领域，具体涉及图像处理、深度学习、计算机视觉等，尤其涉及一种场景分类模型的训练方法、遥感影像场景分类方法、电子设备及存储介质。

背景技术：

1、遥感影像是由高空传感器对地表进行扫描和拍摄而捕获的图像数据，其包含了详细的地物构成和分布信息。但由于遥感场景构成复杂、地物目标较小、地物空间分布杂乱等因素，使得同一类遥感场景图像之间差异较大，而不同类遥感场景图像之间差异较小。另外，遥感影像中通常会包含多个类别的场景，而每个类别的场景在遥感影像中往往只占一小块区域，使得场景的分类更加困难。因此，如何提高遥感影像场景分类的准确性，仍然是一件非常具有挑战性的任务。

技术实现思路

1、本发明提供一种场景分类模型的训练方法、遥感影像场景分类方法、电子设备、存储介质及计算机程序产品。

2、根据本发明的一个方面，提供了一种场景分类模型的训练方法，场景分类模型包括特征提取网络、第一分类网络和第二分类网络，上述方法包括：获取第一样本影像集，第一样本影像集包括多个第一影像块和多个第一影像块各自的样本标签，样本标签用于表征相应的第一影像块中多个待分类场景所属的场景类别；利用特征提取网络对第一样本影像集中的每个第一影像块进行特征提取，得到每个第一影像块的局部特征；其中，特征提取网络是基于第二样本影像集对自动编码器进行训练而得到的；基于每个第一影像块的局部特征，生成针对每个第一影像块的视觉单词直方图；分别使用第一分类网络和第二分类网络基于视觉单词直方图，得到与各个待分类场景对应的第一分类结果和第二分类结果；根据第一分类结果、第二分类结果、样本标签和多个待分类场景之间的关联关系，确定针对第一分类结果的第一损失值和针对第二分类结果的第二损失值；以及根据第一损失值和第二损失值，分别调整第一分类网络和第二分类网络各自的参数。

3、根据本发明的另一方面，提供了一种遥感影像场景分类方法，包括：基于待分类的遥感影像，获取多个第三影像块；利用场景分类模型中的特征提取网络对每个第三影像块进行特征提取，得到每个第三影像块的局部特征；基于每个第三影像块的局部特征，生成针对每个第三影像块的视觉单词直方图；利用场景分类模型中的第一分类网络和第二分类网络分别基于每个视觉单词直方图，得到各个待分类场景对应的第一分类结果和第二分类结果；以及根据各个待分类场景对应的第一分类结果和第二分类结果，得到针对待分类的遥感影像的场景分类结果；其中，场景分类模型是根据以上实施例中描述的场景分类模型的训练方法训练得到的。

4、根据本发明的另一方面，提供了一种电子设备，包括：一个或多个处理器；存储器，用于存储一个或多个程序，其中，当所述一个或多个程序被所述一个或多个处理器执行时，使得一个或多个处理器执行实现如上所述的方法。

5、根据本发明的另一方面，提供了一种计算机可读存储介质，其上存储有可执行指令，该指令被处理器执行时使处理器执行实现如上所述的方法。

6、根据本发明的另一方面，提供了一种计算机程序产品，包括计算机程序，该计算机程序在被处理器执行时实现如上所述的方法。

技术特征：

1.一种场景分类模型的训练方法，所述场景分类模型包括特征提取网络、第一分类网络和第二分类网络，所述方法包括：

2.根据权利要求1所述的方法，其中，所述关联关系包括多个待分类场景之间的相关关系和混淆关系；所述第一分类网络包括与所述多个待分类场景各自对应的第一分类子网络；所述第二分类网络包括与所述多个待分类场景各自对应的第二分类子网络；

3.根据权利要求2所述的方法，其中，响应于所述相关关系包括与第一待分类场景具有相关性的第二待分类场景，针对任意一个第一分类子网络对应的第一待分类场景，执行以下操作中的其中之一：

4.根据权利要求3所述的方法，其中，响应于所述混淆关系包括第三待分类场景易被混淆为第一待分类场景，针对任意一个第二分类子网络对应的第一待分类场景，执行以下操作中的其中之一：

5.根据权利要求4所述的方法，其中，所述根据所述第一损失值和所述第二损失值，分别调整所述第一分类网络和第二分类网络各自的参数包括：

6.根据权利要求5所述的方法，其中，

7.根据权利要求3所述的方法，其中，

8.根据权利要求1至7中任一项所述的方法，还包括：

9.一种遥感影像场景分类方法，包括：

10.一种电子设备，包括：

技术总结
本发明提供了一种场景分类模型的训练方法、遥感影像场景分类方法及设备，具体实现方案包括：获取多个第一影像块和多个第一影像块各自的样本标签；利用特征提取网络对每个第一影像块进行特征提取，得到每个第一影像块的局部特征；基于每个第一影像块的局部特征，生成针对每个第一影像块的视觉单词直方图；分别使用第一分类网络和第二分类网络基于视觉单词直方图，得到与各个待分类场景对应的第一分类结果和第二分类结果；根据第一分类结果、第二分类结果、样本标签和多个待分类场景之间的关联关系，确定针对第一分类结果的第一损失值和针对第二分类结果的第二损失值；根据第一损失值和第二损失值，分别调整第一分类网络和第二分类网络各自的参数。

技术研发人员：鲁锦涛,张丽,李洁,邹圣兵
受保护的技术使用者：北京数慧时空信息技术有限公司
技术研发日：
技术公布日：2024/1/16