[发明专利]实例分割模型的训练方法、装置、系统及存储介质

权利要求书

1.一种实例分割模型的训练方法，包括：

获取训练图像和对应的标注数据，其中所述标注数据包括所述训练图像中目标的真实包围框的位置信息；

对所述训练图像执行仿射变换操作，以获得变换后的训练图像；

将所述训练图像输入所述实例分割模型，以对所述训练图像进行实例分割并获得第一预测掩码；

将所述变换后的训练图像输入所述实例分割模型，以对所述变换后的训练图像进行实例分割并获得第二预测掩码；

对所述第一预测掩码执行所述仿射变换操作，以获得变换后的掩码；

基于所述真实包围框、所述第一预测掩码、所述变换后的掩码和所述第二预测掩码确定主损失函数的函数值；以及

基于所述主损失函数的函数值训练所述实例分割模型。

2.如权利要求1所述的方法，其中，所述将所述训练图像输入所述实例分割模型以对所述训练图像进行实例分割并获得第一预测掩码，包括：

将所述训练图像输入所述实例分割模型，获得所述目标的第一包围框和所述第一包围框的得分；以及

根据所述第一包围框和所述第一包围框的得分，确定所述第一预测掩码。

3.如权利要求2所述的方法，其中，所述方法还包括：

基于所述真实包围框、所述第一包围框和所述第一包围框的得分确定第一辅助损失函数的函数值；

其中，所述训练所述实例分割模型还同时基于所述第一辅助损失函数的函数值。

4.如权利要求1至3任一项所述的方法，其中，

所述将所述变换后的训练图像输入所述实例分割模型以对所述变换后的训练图像进行实例分割并获得第二预测掩码，包括：

将所述变换后的训练图像输入所述实例分割模型，获得所述目标的第二包围框和所述第二包围框的得分；以及

根据所述第二包围框和所述第二包围框的得分，确定所述第二预测掩码；

所述方法还包括：

确定所述变换后的掩码的最小包围框；以及

基于所述最小包围框、所述第二包围框和所述第二包围框的得分确定第二辅助损失函数的函数值；

其中，所述训练所述实例分割模型还同时基于所述第二辅助损失函数的函数值。

5.如权利要求1至3任一项所述的方法，其中，所述基于所述真实包围框、所述第一预测掩码、所述变换后的掩码和所述第二预测掩码确定主损失函数的函数值包括：

基于所述真实包围框和所述第一预测掩码计算第一主导损失函数的函数值；

基于所述变换后的掩码和所述第二预测掩码计算所述第二主导损失函数的函数值；以及

基于所述第一主导损失函数的函数值和所述第二主导损失函数的函数值计算所述主损失函数的函数值。

6.如权利要求5所述的方法，其中，所述基于所述变换后的掩码和所述第二预测掩码计算所述第二主导损失函数的函数值包括：针对每次仿射变换操作且对于所述训练图像中的每个目标，

基于该目标的第i个第一预测掩码经该次仿射变换操作所获得的变换后的掩码以及对所述训练图像执行该次仿射变换操作后进行实例分割所获得的该目标的第j个第二预测掩码计算该目标的子目标函数的函数值；以及

根据该目标的所有子目标函数的函数值利用如下公式计算该目标的第二主导损失函数的函数值：

其中，N表示该目标的第一预测掩码的个数，M表示对所述训练图像执行该次仿射变换操作后进行实例分割所获得的该目标的第二预测掩码的个数。

7.如权利要求6所述的方法，其中，所述子目标函数的函数值根据以下公式计算：

其中，

表示基于投影到x轴所获得的响应函数和投影到x轴所获得的响应函数所计算的x轴损失函数；

表示基于投影到y轴所获得的响应函数和投影到y轴所获得的响应函数所计算的y轴损失函数。

8.如权利要求5所述的方法，其中，所述第二主导损失函数是多实例学习损失函数或像素级别损失函数。