[发明专利]视觉模型的训练方法、装置、设备及存储介质

权利要求书

1.一种目标视觉模型的训练方法，其特征在于，所述方法包括：

获取第一样本图像和背景图像，所述背景图像是指任意场景的图像，所述第一样本图像是指承载了第一目标物体的图像；

根据所述第一样本图像获取目标图像，并将所述目标图像置于所述背景图像得到第二样本图像，所述目标图像是指所述第一样本图像中承载物体的区域的图像；

将各个样本图像输入至待训练模型中，以使得所述待训练模型根据所述样本图像预测出对应的预测类别，其中，所述样本图像包括所述第一样本图像和所述第二样本图像，所述样本图像标注有图像中的物体的类别标签，所述类别标签用于指示所述样本图像中的物体所属的目标类别；

根据所述目标类别和所述预测类别，确定第一损失值，并根据所述第一损失值对所述待训练模型的模型参数进行调整，直至满足预设的停止训练条件，将训练好的所述待训练模型作为目标视觉模型。

2.根据权利要求1所述的目标视觉模型的训练方法，其特征在于，所述待训练模型还包括目标分割分支，所述样本图像还标注有图像中的物体的分割标签，所述分割标签用于指示所述样本图像中的物体的分割数据；

所述根据所述第一损失值对所述待训练模型的模型参数进行调整，直至满足预设的停止训练条件，将训练好的所述待训练模型作为目标视觉模型，包括：

获取所述目标分割分支根据所述样本图像输出的预测分割数据；

根据所述预测分割数据和所述分割标签对应的分割数据，确定第二损失值；

根据所述第一损失值和所述第二损失值，对所述待训练模型的模型参数进行调整，直至满足预设的停止训练条件，将训练好的所述待训练模型作为目标视觉模型。

3.根据权利要求2所述的目标视觉模型的训练方法，其特征在于，所述待训练模型还包括关键点检测分支，所述样本图像还标注有图像中的物体的关键点标签，所述关键点标签用于指示所述样本图像中的物体的关键点信息；

所述根据所述第一损失值和所述第二损失值，对所述待训练模型的模型参数进行调整，直至满足预设的停止训练条件，将训练好的所述待训练模型作为目标视觉模型，包括：

获取所述关键点检测分支根据所述样本图像输出的预测关键点；

根据所述预测关键点和所述关键点标签对应的关键点信息，确定第三损失值；

根据所述第一损失值、所述第二损失值和所述第三损失值，对所述待训练模型的模型参数进行调整，直至满足预设的停止训练条件，将训练好的所述待训练模型作为目标视觉模型。

4.根据权利要求3所述的目标视觉模型的训练方法，其特征在于，所述方法还包括：

获取待检测图像，并将所述待检测图像输入所述目标视觉模型；

获取所述关键点检测分支根据所述待检测图像输出的第二目标物体的关键点，所述第二目标物体是指所述待检测图像中承载的物体。

5.根据权利要求2所述的目标视觉模型的训练方法，其特征在于，所述方法还包括：

获取待识别图像，并将所述待识别图像输入所述目标视觉模型；

获取所述目标分割分支根据所述待识别图像输出的第三目标物体的分割数据，所述第三目标物体是指所述待识别图像中承载的物体。

6.根据权利要求1所述的目标视觉模型的训练方法，其特征在于，所述第二样本图像对应的类别标签与所述第一样本图像对应的类别标签相同。

7.根据权利要求1-6任一项所述的目标视觉模型的训练方法，其特征在于，所述样本图像还包括第三样本图像，所述第三样本图像通过以下步骤得到：

对待处理图像进行图像预处理，得到所述第三样本图像，所述待处理图像包括第一样本图像、第二样本图像中的至少一种，所述图像预处理包括图像旋转、图像缩放、图像裁剪、图像滤波、图像填充、图像对比度变换、图像颜色变换、图像锐化操作中的至少一种。