[发明专利]目标检测网络的训练方法、目标检测方法及装置

权利要求书

1.一种目标检测方法，其特征在于，包括：

读取测试图像；

基于预先训练的目标检测网络对所述测试图像进行目标检测，确定所述测试图像对应的目标检测结果；其中，所述目标检测结果包括所述测试图像中的前景物体，以及所述前景物体的类别信息；

所述目标检测网络通过以下方法进行训练：

基于初始检测网络确定第一检测网络和第二检测网络，其中，所述第二检测网络的网络规模高于所述第一检测网络；

基于第一样本数据集对所述第一检测网络进行预训练，基于第二样本数据集对预训练后的第一检测网络进行微调训练；以及，基于所述第二样本数据集对所述第二检测网络进行训练；其中，所述第一样本数据集的样本数量多于所述第二样本数据集，所述第一样本数据集与所述第二样本数据集所包含的物体类别相同，所述包含的样本数据不同；

基于训练好的所述第一检测网络的输出值，和训练好的所述第二检测网络的输出值，对所述第一检测网络进行蒸馏训练，得到目标检测网络；其中，所述第二样本数据集与所述目标检测网络的测试集之间的相似度，高于所述第一样本数据集与所述测试集之间的相似度。

2.根据权利要求1所述的方法，其特征在于，基于初始检测网络确定第一检测网络和第二检测网络，包括：

通过对所述初始检测网络进行规模调整，或者通过对所述初始检测网络进行规模调整和网络结构调整，确定所述第一检测网络和所述第二检测网络；

所述规模调整包括对所述初始检测网络的网络宽度按照第一预设比例进行调整和/或对所述初始检测网络的网络深度按照第二预设比例进行调整。

3.根据权利要求1或2所述的方法，其特征在于，所述基于训练好的所述第一检测网络的输出值，和训练好的所述第二检测网络的输出值，对所述第一检测网络进行蒸馏训练，包括：

基于所述第一检测网络的第一融合模块的输出值和所述第二检测网络的第二融合模块的输出值，确定损失值，其中，所述损失值表征所述第一融合模块的输出值和所述第二融合模块的输出值之间的差异；

基于所述损失值和所述第二检测网络对所述第一检测网络进行蒸馏训练。

4.根据权利要求1所述的方法，其特征在于，所述对所述第一检测网络进行蒸馏训练之后，所述方法还包括：

对蒸馏训练后的所述第一检测网络进行模型量化处理。

5.根据权利要求4所述的方法，其特征在于，所述基于预先训练的目标检测网络对所述测试图像进行目标检测，确定所述测试图像对应的目标检测结果，包括：

基于所述目标检测网络对所述测试图像进行目标检测，确定所述测试图像包含的物体属于各个类别的置信度信息，以及各个物体对应的前景置信度信息，其中，所述前景置信度信息用于表示各个物体为前景的概率；

基于所述前景置信度信息，筛选出前景物体；

基于所述前景物体的置信度信息，确定所述测试图像对应的目标检测结果。

6.根据权利要求5所述的方法，其特征在于，所述目标检测网络对所述测试图像进行目标检测后，检测结果还包括各个物体对应的位置信息；

所述基于所述前景物体的置信度信息，确定所述测试图像对应的目标检测结果，包括：

基于所述前景物体的置信度信息中的最高置信度和所述位置信息，确定所述测试图像对应的目标检测结果。