[发明专利]模型训练方法、装置、电子设备和存储介质

权利要求书

1.一种模型训练方法，其特征在于，包括：

获取第一样本；

对所述第一样本中符合裁剪条件的样本进行裁剪，得到第二样本；

根据包括所述第一样本和所述第二样本的训练样本进行模型训练，得到第一模型；

根据与所述第一模型输出的预测结果对应的调整策略调整所述第一模型，以得到第二模型，所述预测结果对应于所述训练样本。

2.根据权利要求1所述的方法，其特征在于，所述第一样本和所述第二样本为遥感图像，所述第一模型和所述第二模型用于检测遥感图像中是否包含预设类别的地物。

3.根据权利要求2所述的方法，其特征在于，所述对所述第一样本中符合裁剪条件的样本进行裁剪，得到第二样本，包括：

获取所述第一样本包含的地物所属的类别；

若所述地物所属的类别为所述预设类别，则确定与所述预设类别对应的裁剪尺寸；

根据所述裁剪尺寸裁剪所述第一样本，得到所述第二样本。

4.根据权利要求2所述的方法，其特征在于，所述对所述第一样本中符合裁剪条件的样本进行裁剪，得到第二样本，包括：

确定所述第一样本包含的地物在所述第一样本中的尺寸信息；

若所述尺寸信息符合预设尺寸范围，则对所述第一样本进行裁剪，得到所述第二样本。

5.根据权利要求3所述的方法，其特征在于，所述根据与所述第一模型输出的预测结果对应的调整策略调整所述第一模型，包括：

获取所述第一模型输出的预测结果的置信度；

若所述置信度小于预设阈值，则根据与所述置信度具有负相关关系的第一损失函数调整所述第一模型；

若所述置信度大于或等于所述预设阈值，则根据第二损失函数调整所述第一模型。

6.一种模型训练装置，其特征在于，包括：

获取模块，用于获取第一样本；

裁剪模块，用于对所述第一样本中符合裁剪条件的样本进行裁剪，得到第二样本；

训练模型，用于根据包括所述第一样本和所述第二样本的训练样本进行模型训练，得到第一模型；

调整模型，用于根据与所述第一模型输出的预测结果对应的调整策略调整所述第一模型，以得到第二模型，所述预测结果对应于所述训练样本。

7.根据权利要求6所述的装置，其特征在于，所述裁剪模块具体包括：

获取单元，用于获取所述第一样本包含的地物所属的类别；

确定单元，用于若所述地物所属的类别为所述预设类别，则确定与所述预设类别对应的裁剪尺寸；

裁剪单元，用于根据所述裁剪尺寸裁剪所述第一样本，得到所述第二样本。

8.根据权利要求7所述的装置，其特征在于，所述调整模块具体包括：

获取单元，用于获取所述第一模型输出的预测结果的置信度；

第一调整单元，用于若所述置信度小于预设阈值，则根据与所述置信度具有负相关关系的第一损失函数调整所述第一模型；

第二调整单元，用于若所述置信度大于或等于所述预设阈值，则根据第二损失函数调整所述第一模型。

9.一种电子设备，其特征在于，包括：处理器和存储器；其中，所述存储器用于存储一条或多条计算机指令，其中，所述一条或多条计算机指令被所述处理器执行时实现：

获取第一样本；

对所述第一样本中符合裁剪条件的样本进行裁剪，得到第二样本；

根据包括所述第一样本和所述第二样本的训练样本进行模型训练，得到第一模型；

根据与所述第一模型输出的预测结果对应的调整策略调整所述第一模型，以得到第二模型，所述预测结果对应于所述训练样本。

10.一种存储计算机指令的计算机可读存储介质，其特征在于，当所述计算机指令被一个或多个处理器执行时，致使所述一个或多个处理器至少执行以下的动作：

获取第一样本；

对所述第一样本中符合裁剪条件的样本进行裁剪，得到第二样本；

根据包括所述第一样本和所述第二样本的训练样本进行模型训练，得到第一模型；

根据与所述第一模型输出的预测结果对应的调整策略调整所述第一模型，以得到第二模型，所述预测结果对应于所述训练样本。