[发明专利]模型的训练方法、装置、设备、存储介质及图像检测方法

权利要求书

1.一种模型的训练方法，包括：

将第一输入对象输入预先训练的教师网络，得到第一预测框；以及，将第二输入对象输入待训练的学生网络，得到第二预测框；

根据所述第一预测框和所述第二预测框之间的差异，调整所述学生网络的参数，得到目标检测模型；

其中，所述目标检测模型用于识别待检测图像中的3D物体信息。

2.根据权利要求1所述的方法，其中，所述第一预测框和所述第二预测框之间的差异包括交并比，所述第二预测框为多个；

根据所述第一预测框和所述第二预测框之间的差异，调整所述学生网络的参数，包括：

根据各所述第二预测框与对应的第一预测框之间的交并比，从多个所述第二预测框中确定出至少一个相似预测框；其中，所述相似预测框与对应的第一预测框之间的交并比大于交并比阈值；

根据各所述相似预测框对应的交并比，按照从大到小的顺序依次调整所述学生网络的参数。

3.根据权利要求1所述的方法，其中，根据所述第一预测框和所述第二预测框之间的差异，调整所述学生网络的模型参数之前，还包括：

利用所述教师网络的特征提取层提取所述第一输入对象的第一特征；

以及，在将所述第二输入对象输入待训练的学生网络之后，还包括：

利用所述学生网络的特征提取层提取所述第二输入对象的第二特征；

根据所述第一特征和所述第二特征之间的差异，调整所述学生网络的特征提取层的参数。

4.根据权利要求3所述的方法，其中，所述第一特征和所述第二特征之间的差异包括余弦距离。

5.根据权利要求1至4任一项所述的方法，其中，所述第一输入对象为样本图像的深度图；所述第二输入对象为所述样本图像的RGB图。

6.一种图像的检测方法，包括：

将待检测图像输入目标检测模型；

接收所述目标检测模型输出的所述待检测图像中的3D物体信息；

其中，所述目标检测模型采用根据权利要求1至5任一项所述的模型训练方法得到。

7.根据权利要求6所述的方法，其中，所述3D物体信息包括分类信息、位置信息、尺寸信息以及角度信息中的至少一项。

8.一种模型的训练装置，包括：

第一预测框获取模块，用于将第一输入对象输入预先训练的教师网络，得到第一预测框；

第二预测框获取模块，用于将第二输入对象输入待训练的学生网络，得到第二预测框；

参数调整模块，用于根据所述第一预测框和所述第二预测框之间的差异，调整所述学生网络的参数，得到目标检测模型；

其中，所述目标检测模型用于识别待检测图像中的3D物体信息。

9.根据权利要求8所述的装置，其中，所述第一预测框和所述第二预测框之间的差异包括交并比，所述第二预测框为多个；

所述参数调整模块包括：

相似预测框确定单元，根据各所述第二预测框与对应的第一预测框之间的交并比，从多个所述第二预测框中确定出至少一个相似预测框；其中，所述相似预测框与对应的第一预测框之间的交并比大于交并比阈值；

参数调整单元，用于根据各所述相似预测框对应的交并比，按照从大到小的顺序依次调整所述学生网络的参数。

10.根据权利要求8所述的装置，还包括：

第一特征提取模块，用于利用所述教师网络的特征提取层提取所述第一输入对象的第一特征；

第二特征提取模块，用于利用所述学生网络的特征提取层提取所述第二输入对象的第二特征；

特征提取层参数调整模块，用于根据所述第一特征和所述第二特征之间的差异，调整所述学生网络的特征提取层的参数。

11.根据权利要求10所述的装置，其中，所述第一特征和所述第二特征之间的差异包括余弦距离。