[发明专利]用于检测物品类别和位置的检测网络的训练方法和装置

权利要求书

1.一种用于检测物品类别和位置的检测网络的训练方法，其特征在于，包括：

生成物品的虚拟图片；

通过图像处理网络对所述虚拟图片进行处理，以生成与所述物品的虚拟图片对应的渲染图片，所述渲染图片与所述虚拟图片具有相同的标注信息；

利用所述渲染图片及所述标注信息，对用于检测物品类别和位置的检测网络进行训练，得到训练后的所述检测网络。

2.根据权利要求1所述的方法，其特征在于，所述图像处理网络为对抗式生成网络，

通过图像处理网络对所述虚拟图片进行处理的步骤之前，包括：

在所述对抗式生成网络的第一损失函数的基础上，构建所述对抗式生成网络的第二损失函数，并使用所述第二损失函数训练所述对抗式生成网络，其中，所述第一损失函数为所述对抗式生成网络的原有损失函数；

所述第二损失函数根据以下各项构建：所述第一损失函数、所述渲染图片与所述虚拟图片的损失、所述渲染图片的标注信息与所述虚拟图片的标注信息的损失。

3.根据权利要求1所述的方法，其特征在于，生成物品的虚拟图片的步骤，包括：

通过2D建模生成所述物品的虚拟图片；或者，

通过3D建模生成所述物品的2D图片和3D深度图，并将所述2D图片作为所述物品的虚拟图片。

4.一种用于检测物品类别和位置的检测网络的训练装置，其特征在于，包括：

虚拟图片生成模块，用于生成物品的虚拟图片；

虚拟图片处理模块，用于通过图像处理网络对所述虚拟图片进行处理，以生成与所述物品的虚拟图片对应的渲染图片，所述渲染图片与所述虚拟图片具有相同的标注信息；

第一网络训练模块，用于利用所述渲染图片及所述标注信息，对用于检测物品类别和位置的检测网络进行训练，得到训练后的所述检测网络。

5.根据权利要求4所述的装置，其特征在于，所述图像处理网络为对抗式生成网络，

所述装置还包括：

第二网络训练模块，用于在所述对抗式生成网络的第一损失函数的基础上，构建所述对抗式生成网络的第二损失函数，并使用所述第二损失函数训练所述对抗式生成网络，其中，所述第一损失函数为所述对抗式生成网络的原有损失函数；

所述第二损失函数根据以下各项构建：所述第一损失函数、所述渲染图片与所述虚拟图片的损失、所述渲染图片的标注信息与所述虚拟图片的标注信息的损失。

6.根据权利要求4所述的装置，其特征在于，所述虚拟图片生成模块还用于：

通过2D建模生成所述物品的虚拟图片；或者，

通过3D建模生成所述物品的2D图片和3D深度图，并将所述2D图片作为所述物品的虚拟图片。

7.一种电子设备，其特征在于，包括：

一个或多个处理器；

存储器，用于存储一个或多个程序，

当所述一个或多个程序被所述一个或多个处理器执行时，使得所述一个或多个处理器实现如权利要求1-3中任一所述的方法。

8.一种计算机可读介质，其上存储有计算机程序，其特征在于，所述程序被处理器执行时实现如权利要求1-3中任一所述的方法。