一种残膜回收机防缠绕挑膜装置的制 一种秧草收获机用电力驱动行走机构

一种目标检测模型的训练方法、目标检测方法及装置与流程

2022-08-13 14:16:03 来源:中国专利 TAG:


1.本公开涉及计算机技术领域,尤其涉及一种目标检测模型的训练方法、目标检测方法及装置。


背景技术:

2.目前,在目标检测等领域,电子设备可以将某一目标检测图片输入到已经训练完成的神经网络模型中,进而该神经网络模型可以预测出该目标检测图片中包括的多个目标(或物体)以及该多个目标各自的类别。
3.但是,上述神经网络模型的训练过程中,可能需要大量高质量的目标检测训练样本,具体为需要大量的人力物力去标注每个目标检测训练样本中包括的各个目标的类别等,如此可能会影响神经网络模型的训练效率,进而降低神经网络模型的预测效率。


技术实现要素:

4.本公开提供一种目标检测模型的训练方法、目标检测方法及装置,解决了相关技术中需要大量的人力物力去标注每个目标检测训练样本中包括的各个目标的类别,可能会影响神经网络模型的训练效率,进而降低神经网络模型的预测效率的技术问题。
5.本公开实施例的技术方案如下:
6.根据本公开实施例的第一方面,提供一种目标检测模型的训练方法。该方法可以包括:确定已检测图片包括的多个检测框中每一个检测框的初始特征以及该每一个检测框的初始特征对应的至少一个权重特征,其中,一个检测框的初始特征对应的至少一个权重特征中每一个权重特征与该检测框的初始特征之间的相似度大于或等于相似度阈值,该每一个检测框的初始特征对应的至少一个权重特征为第一分类器中包括的权重的特征,一个权重对应该第一分类器中的一个类别;基于该多个检测框中每一个检测框的初始特征以及该每一个检测框的初始特征对应的至少一个权重特征,生成该每一个检测框的目标特征;基于该每一个检测框的目标特征,对第一目标检测模型训练,以生成第二目标检测模型。
7.可选地,上述第一目标检测模型中包括第一子网络模型和第二子网络模型,上述确定已检测图片包括的多个检测框中每一个检测框的初始特征具体包括:获取该已检测图片;将该已检测图片输入该第一子网络模型,进行特征识别,得到该已检测图片的初始特征;将该已检测图片的初始特征输入该第二子网络模型,得到该多个检测框中每一个检测框的位置以及该每一个检测框的大小;基于该每一个检测框的位置以及该每一个检测框的大小,从该已检测图片的初始特征中提取该每一个检测框的初始特征。
8.可选地,上述第一分类器为第一图片分类模型中包括的分类器,该第一图片分类模型中包括第三子网络模型,该第三子网络模型中的参数与该第一子网络模型中的参数相同,上述目标检测模型的训练方法还包括:获取多个分类图片;将该多个分类图片输入该第三子网络模型,得到该多个分类图片中每个分类图片的初始特征;将该每个分类图片的初始特征输入该第一分类器,得到该每个分类图片在该第一分类器中的预测分类结果。
9.可选地,上述目标检测模型的训练方法还包括:确定第一损失和第二损失,该第一损失用于表征多个分类图片中每个分类图片的真实分类结果与该每个分类图片在该第一分类器中的预测分类结果之间的不一致程度,该第二损失用于表征该多个检测框中每一个检测框的空间不一致程度,其中,一个检测框的空间不一致程度包括该检测框的真实位置与该检测框在该第一目标检测模型中的预测位置之间的不一致程度,以及该检测框的真实大小与该检测框在该第一目标检测模型中的预测大小之间的不一致程度;上述基于该每一个检测框的目标特征,对第一目标检测模型训练,以生成第二目标检测模型具体包括:将该每一个检测框的目标特征输入第二分类器,得到该每一个检测框在该第二分类器中的预测分类结果,该第二分类器为该第一目标检测模型中包括的分类器;确定第三损失和第四损失,该第三损失用于表征该每一个检测框的真实分类结果与该每一个检测框在该第二分类器中的预测分类结果之间的不一致程度,该第四损失用于表征该每一个检测框中是否包括物体的真实结果与该每一个检测框在该第二分类器中被预测为是否包括物体的结果之间的不一致程度;根据该第一损失、该第二损失、该第三损失以及该第四损失,确定目标损失;基于该目标损失,更新该第一目标检测模型中的参数,以生成该第二目标检测模型。
10.可选地,上述目标检测模型的训练方法还包括:将第一检测框的目标特征确定为预设均值,该第一检测框为该多个检测框中的一个;基于该预设均值和预设标准差,生成该第一检测框的多个拓展特征;基于该多个检测框中每一个检测框的多个拓展特征,对该第一目标检测模型进行训练。
11.根据本公开实施例的第二方面,提供一种目标检测方法。该方法可以包括:获取待检测图片;将该待检测图片输入第二目标检测模型,得到该待检测图片包括的至少一个检测框各自的类别,该第二目标检测模型是基于上述第一方面中任一种可选地目标检测模型的训练方法训练得到的。
12.可选地,上述第二目标检测模型中包括第一子网络模型、第二子网络模型以及第二分类器,上述将该待检测图片输入第二目标检测模型,得到该待检测图片包括的至少一个检测框各自的类别具体包括:将该待检测图片输入该第一子网络模型,进行特征识别,得到该待检测图片的初始特征;将该待检测图片的初始特征输入该第二子网络模型,得到该至少一个检测框各自的位置以及该至少一个检测框各自的大小;基于该至少一个检测框各自的位置以及该至少一个检测框各自的大小,从该待检测图片的初始特征中提取该至少一个检测框各自的初始特征;将该至少一个检测框各自的初始特征输入该第二分类器,得到该至少一个检测框各自的类别。
13.根据本公开实施例的第三方面,提供一种目标检测模型的训练装置。该装置可以包括:确定模块和处理模块;该确定模块,被配置为确定已检测图片包括的多个检测框中每一个检测框的初始特征以及该每一个检测框的初始特征对应的至少一个权重特征,其中,一个检测框的初始特征对应的至少一个权重特征中每一个权重特征与该检测框的初始特征之间的相似度大于或等于相似度阈值,该每一个检测框的初始特征对应的至少一个权重特征为第一分类器中包括的权重的特征,一个权重对应该第一分类器中的一个类别;该处理模块,被配置为基于该多个检测框中每一个检测框的初始特征以及该每一个检测框的初始特征对应的至少一个权重特征,生成该每一个检测框的目标特征;该处理模块,还被配置为基于该每一个检测框的目标特征,对第一目标检测模型训练,以生成第二目标检测模型。
14.可选地,上述第一目标检测模型中包括第一子网络模型和第二子网络模型,上述目标检测模型的训练装置还包括获取模块;该获取模块,被配置为获取该已检测图片;该处理模块,具体被配置为将该已检测图片输入该第一子网络模型,进行特征识别,得到该已检测图片的初始特征;该处理模块,具体还被配置为将该已检测图片的初始特征输入该第二子网络模型,得到该多个检测框中每一个检测框的位置以及该每一个检测框的大小;该处理模块,具体还被配置为基于该每一个检测框的位置以及该每一个检测框的大小,从该已检测图片的初始特征中提取该每一个检测框的初始特征。
15.可选地,上述第一分类器为第一图片分类模型中包括的分类器,该第一图片分类模型中包括第三子网络模型,该第三子网络模型中的参数与该第一子网络模型中的参数相同;该获取模块,还被配置为获取多个分类图片;该处理模块,还被配置为将该多个分类图片输入该第三子网络模型,得到该多个分类图片中每个分类图片的初始特征;该处理模块,还被配置为将该每个分类图片的初始特征输入该第一分类器,得到该每个分类图片在该第一分类器中的预测分类结果。
16.可选地,该确定模块,还被配置为确定第一损失和第二损失,该第一损失用于表征多个分类图片中每个分类图片的真实分类结果与该每个分类图片在该第一分类器中的预测分类结果之间的不一致程度,该第二损失用于表征该多个检测框中每一个检测框的空间不一致程度,其中,一个检测框的空间不一致程度包括该检测框的真实位置与该检测框在该第一目标检测模型中的预测位置之间的不一致程度,以及该检测框的真实大小与该检测框在该第一目标检测模型中的预测大小之间的不一致程度;该处理模块,具体被配置为将该每一个检测框的目标特征输入第二分类器,得到该每一个检测框在该第二分类器中的预测分类结果,该第二分类器为该第一目标检测模型中包括的分类器;该确定模块,具体被配置为确定第三损失和第四损失,该第三损失用于表征该每一个检测框的真实分类结果与该每一个检测框在该第二分类器中的预测分类结果之间的不一致程度,该第四损失用于表征该每一个检测框中是否包括物体的真实结果与该每一个检测框在该第二分类器中被预测为是否包括物体的结果之间的不一致程度;该确定模块,具体还被配置为根据该第一损失、该第二损失、该第三损失以及该第四损失,确定目标损失;该处理模块,具体还被配置为基于该目标损失,更新该第一目标检测模型中的参数,以生成该第二目标检测模型。
17.可选地,该确定模块,还被配置为将第一检测框的目标特征确定为预设均值,该第一检测框为该多个检测框中的一个;该处理模块,还被配置为基于该预设均值和预设标准差,生成该第一检测框的多个拓展特征;该处理模块,还被配置为基于该多个检测框中每一个检测框的多个拓展特征,对该第一目标检测模型进行训练。
18.根据本公开实施例的第四方面,提供一种目标检测装置。该装置可以包括:获取模块和处理模块;该获取模块,被配置为获取待检测图片;该处理模块,被配置为将待检测图片输入第二目标检测模型,得到该待检测图片包括的至少一个检测框各自的类别,该第二目标检测模型是基于上述第一方面中任一种可选地目标检测模型的训练方法训练得到的。
19.可选地,该第二目标检测模型中包括第一子网络模型、第二子网络模型以及第二分类器;该处理模块,具体被配置为将该待检测图片输入该第一子网络模型,进行特征识别,得到该待检测图片的初始特征;该处理模块,具体还被配置为将该待检测图片的初始特征输入该第二子网络模型,得到该至少一个检测框各自的位置以及该至少一个检测框各自
的大小;该处理模块,具体还被配置为基于该至少一个检测框各自的位置以及该至少一个检测框各自的大小,从该待检测图片的初始特征中提取该至少一个检测框各自的初始特征;该处理模块,具体还被配置为将该至少一个检测框各自的初始特征输入该第二分类器,得到该至少一个检测框各自的类别。
20.根据本公开实施例的第五方面,提供一种电子设备,可以包括:处理器和被配置为存储处理器可执行指令的存储器;其中,处理器被配置为执行所述指令,以实现上述第一方面中任一种可选地目标检测模型的训练方法,或者实现上述第二方面中任一种可选地目标检测方法。
21.根据本公开实施例的第六方面,提供一种计算机可读存储介质,计算机可读存储介质上存储有指令,当该计算机可读存储介质中的指令由电子设备的处理器执行时,使得该电子设备能够执行上述第一方面中任一种可选地目标检测模型的训练方法,或者执行上述第二方面中任一种可选地目标检测方法。
22.根据本公开实施例的第七方面,提供一种计算机程序产品,该计算机程序产品包括计算机指令,当该计算机指令在电子设备的处理器上运行时,使得该电子设备执行如第一方面中任一种可选地目标检测模型的训练方法,或者执行如第二方面中任一种可选地目标检测方法。
23.本公开的实施例提供的技术方案至少带来以下有益效果:
24.基于上述任一方面,本公开中,电子设备可以确定已检测图片包括的多个检测框中每一个检测框的初始特征以及该每一个检测框对应的至少一个权重特征,并且基于该每一个检测框的初始特征以及该每一个检测框对应的至少一个权重特征,生成该每一个检测框的目标特征。由于该每一个检测框对应的至少一个权重特征为第一分类器中包括的权重的特征,一个权重对应该第一分类器中的一个类别,而一个检测框对应的至少一个权重特征与该检测框之间的相似度大于或等于相似度阈值,因此,电子设备可以从该第一分类器包括的多个类别中,确定与该检测框的类别(具体为该检测框中包括的物体的类别)较为相似的至少一个类别(即该检测框的初始特征对应的至少一个权重特征各自对应的类别)。然后电子设备可以将该检测框的类别对应的特征(即该检测框的初始特征)与该至少一个类别对应的特征(即该检测框的初始特征对应的至少一个权重特征)相结合,可以实现从分类图片到检测图片上的类别的迁移,无需对目标检测模型的训练样本进行大量的人工标注,可以准确地生成能够表征该每一个检测框的类别(具体为该每一个检测框中包括的物体的类别)的真实数据分布的目标特征。然后,该电子设备可以基于该每一个检测框的目标特征对第一目标检测模型训练,以生成第二目标检测模型,能够提升目标检测模型的训练效率。进而,电子设备可以基于该第二目标检测模型快速、准确地预测出每一个待检测图片中包括的多个物体的类别,即能够提升目标检测模型的预测效率。
25.应当理解的是,以上的一般描述和后文的细节描述仅是示例性和解释性的,并不能限制本公开。
附图说明
26.此处的附图被并入说明书中并构成本说明书的一部分,示出了符合本公开的实施例,并与说明书一起用于解释本公开的原理,并不构成对本公开的不当限定。
27.图1示出了本公开实施例提供的一种目标检测模型的训练方法的流程示意图;
28.图2示出了本公开实施例提供的又一种目标检测模型的训练方法的流程示意图;
29.图3示出了本公开实施例提供的又一种目标检测模型的训练方法的流程示意图;
30.图4示出了本公开实施例提供的又一种目标检测模型的训练方法的流程示意图;
31.图5示出了本公开实施例提供的又一种目标检测模型的训练方法的流程示意图;
32.图6示出了本公开实施例提供的一种目标检测方法的流程示意图;
33.图7示出了本公开实施例提供的又一种目标检测方法的流程示意图;
34.图8示出了本公开实施例提供的一种目标检测模型的训练装置的结构示意图;
35.图9示出了本公开实施例提供的又一种目标检测模型的训练装置的结构示意图;
36.图10示出了本公开实施例提供的一种目标检测装置的结构示意图;
37.图11示出了本公开实施例提供的又一种目标检测装置的结构示意图。
具体实施方式
38.为了使本领域普通人员更好地理解本公开的技术方案,下面将结合附图,对本公开实施例中的技术方案进行清楚、完整地描述。
39.需要说明的是,本公开的说明书和权利要求书及上述附图中的术语“第一”、“第二”等是用于区别类似的对象,而不必用于描述特定的顺序或先后次序。应该理解这样使用的数据在适当情况下可以互换,以便这里描述的本公开的实施例能够以除了在这里图示或描述的那些以外的顺序实施。以下示例性实施例中所描述的实施方式并不代表与本公开相一致的所有实施方式。相反,它们仅是与如所附权利要求书中所详述的、本公开的一些方面相一致的装置和方法的例子。
40.还应当理解的是,术语“包括”指示所描述特征、整体、步骤、操作、元素和/或组件的存在,但并不排除一个或多个其他特征、整体、步骤、操作、元素和/或组件的存在或添加。
41.需要说明的是,本公开所涉及的用户信息(包括但不限于用户设备信息、用户个人信息、用户行为信息等)和数据(包括但不限于已检测图片、分类图片以及待检测图片等),均为经用户授权或者经过各方充分授权的信息和数据。
42.相关技术中,在进行神经网络模型的训练的过程中,可能需要大量高质量的目标检测训练样本,具体为需要大量的人力物力去标注每个目标检测训练样本中包括的各个目标的类别等,可能会影响神经网络模型的训练效率,进而降低神经网络模型的预测效率。
43.基于此,本公开实施例提供一种目标检测模型的训练方法,电子设备可以将已检测图片包括的多个检测框中某一个检测框的类别对应的特征(即该检测框的初始特征)与第一分类器中包括的至少一个类别对应的特征(即该检测框的初始特征对应的至少一个权重特征)相结合,可以实现从分类图片到检测图片上的类别的迁移,无需对目标检测模型的训练样本进行大量的人工标注,可以准确地生成能够表征每一个检测框的类别(具体为该每一个检测框中包括的物体的类别)的真实数据分布的目标特征。并且电子设备可以基于该每一个检测框的目标特征对第一目标检测模型训练,以生成第二目标检测模型,能够提升目标检测模型的训练效率。进而,电子设备可以基于该第二目标检测模型快速、准确地预测出每一个待检测图片中包括的多个物体的类别,即能够提升目标检测模型的预测效率。
44.本公开实施例提供的目标检测模型的训练方法、目标检测方法及装置,应用于目
标检测场景(具体为需要确定出某一个待检测图片包括的每一个物体(或目标)的类别的场景)中。当电子设备确定出已检测图片包括的多个检测框中每一个检测框的初始特征以及该每一个检测框的初始特征对应的至少一个权重特征时,可以依据本公开实施例提供的方法,对第一目标检测模型(可以理解为初始状态下的目标检测模型)训练以生成第二目标检测模型(可以理解为已经训练完成的目标检测模型)。进而,当该电子设备获取到待检测图片时,可以将该待检测图片输入该第二目标检测模型,以得到该待检测图片包括的至少一个检测框各自的类别(也可以理解为该至少一个检测框中各自包括的物体或目标的类别)。
45.以下结合附图对本公开实施例提供的目标检测模型的训练方法以及目标检测方法进行示例性说明:
46.示例性的,执行本公开实施例提供的目标检测模型的训练方法以及目标检测方法的电子设备可以是手机、平板电脑、桌面型、膝上型、手持计算机、笔记本电脑、超级移动个人计算机(ultra-mobile personal computer,umpc)、上网本,以及蜂窝电话、个人数字助理(personal digital assistant,pda)、增强现实(augmented reality,ar)\虚拟现实(virtual reality,vr)设备等可以安装并使用内容社区应用的设备,本公开对该电子设备的具体形态不作特殊限制。其可以与用户通过键盘、触摸板、触摸屏、遥控器、语音交互或手写设备等一种或多种方式进行人机交互。
47.如图1所示,本公开实施例提供的目标检测模型的训练方法可以包括s101-s103。
48.s101、电子设备确定已检测图片包括的多个检测框中每一个检测框的初始特征以及每一个检测框的初始特征对应的至少一个权重特征。
49.其中,一个检测框的初始特征对应的至少一个权重特征中每一个权重特征与该检测框的初始特征之间的相似度大于或等于相似度阈值,该每一个检测框的初始特征对应的权重特征为第一分类器中包括的权重的特征,一个权重对应该第一分类器中的一个类别。
50.应理解,上述多个检测框中的一个检测框用于表征该已检测图片中的一个区域,该检测框(或该区域)中可能包括一个物体(或目标),也可能不包括物体。当该检测框中包括某一个物体时,该物体的类别即为该检测框的类别;当该检测框中不包括物体时,说明该检测框表征的是该已检测图片中的某一背景区域。
51.可选地,本公开实施例中的检测框可以为提议框(proposal)。
52.可以理解的是,上述第一分类器为第一图片分类模型中包括的分类器,该第一图片分类模型为某一初始状态下的图片分类模型,该图片分类模型用于对某一分类图片进行分类,具体为确定(或预测)该分类图片对应的类别。
53.需要说明的是,本公开实施例中的检测图片和分类图片为不同的图片,并且分别作用于不同的网络模型中。
54.具体的,检测图片一般比较大,该检测图片表征出的场景比较复杂,该检测图片中可以包括多个物体(或目标),对于该多个物体中的每一个物体而言,均可以为其配置一个检测框,进而将该每一个物体独立起来。电子设备将该检测图片输入某一目标检测模型,即可以得到该每一个检测框的类别,对于包括有物体的检测框而言,该检测框的类别结尾该检测框中包括的物体的类别。
55.而分类图片一般比较小,该分类图片表征出的场景也比较简单,该分类图片中一般只包括一个物体。电子设备将该分类图片输入到某一图片分类模型,即可以得到该分类
图片的类别,该分类图片的类别即为该分类图片中包括的物体的类别。
56.本公开实施例中,某一个图片分类模型(例如上述第一图片分类模型)中可以包括一个分类器(即第一分类器),该第一分类器可以为多元分类器,即可以将多个分类图片划分为至少三种的类别。
57.应理解,该第一分类器中包括多个类别,一个类别对应一个权重,电子设备可以获取该多个类别中每一个类别对应的权重的特征,然后确定该每一个类别对应的权重的特征与上述多个检测框中某一个检测框的初始特征之间的相似度。当该检测框的初始特征与某一个类别对应的权重的特征之间的相似度大于或等于相似度阈值时,说明该检测框的类别(具体为该检测框中包括的物体的类别)与该类别较为相似,此时电子设备可以将该类别对应的权重的特征,确定为该检测框的初始特征对应的至少一个权重特征中的一个。即对于上述一个检测框的初始特征对应的至少一个权重特征中每一个权重特征而言,该每一个权重特征对应的类别与该检测框的类别较为相似。
58.在公开实施例的一种实现方式中,在上述电子设备确定该多个类别中每一个类别对应的权重的特征与某一个检测框的初始特征之间的相似度之后,还可以把相似度最大的x(x≥1)个类别对应的权重的特征,确定为该检测框的初始特征对应的至少一个权重特征。
59.可选地,电子设备可以余弦距离的形式表征上述每一个类别对应的权重的特征与多个检测框中某一个检测框的初始特征之间的相似度。具体的,当该检测框的初始特征与某一个类别对应的权重的特征之间的余弦距离小于或等于距离阈值时,电子设备可以将该类别对应的权重的特征确定为该检测框的初始特征对应的至少一个权重特征中的一个。或者,该电子设备还可以确定该多个类别中每一个类别对应的权重的特征与该检测框的初始特征之间的余弦距离,然后将余弦距离最小的x个类别对应的权重的特征,确定为该检测框的初始特征对应的至少一个权重特征。
60.可选地,上述第一分类器可以为20000类分类器。
61.s102、电子设备基于多个检测框中每一个检测框的初始特征以及每一个检测框的初始特征对应的至少一个权重特征,生成每一个检测框的目标特征。
62.结合上述实施例的描述,应理解,对于该多个检测框中的某一个检测框而言,该检测框的初始特征对应的至少一个权重特征,为与该检测框的类别(具体为该检测框中包括的物体的类别)较为相似的类别对应的权重的特征。
63.本公开实施例中,电子设备基于该每一个检测框的初始特征以及该每一个检测框的初始特征对应的至少一个权重特征,生成该每一个检测框的目标特征,即可以将该每一个检测框的类别的特征与至少一个类别(具体为图片分类模型中与该每一个检测框的类别较为相似的类别)的特征进行特征融合。由于图片分类模型中包括大量的类别,如此电子设备可以实现从分类图片到检测图片上的类别的迁移,无需对目标检测模型的训练样本进行大量的人工标注,可以准确地生成能够表征该每一个检测框的类别(具体为该每一个检测框中包括的物体的类别)的真实数据分布的目标特征。
64.在一种可选的实现方式中,对于上述多个检测框中的某一个检测框而言,电子设备可以将该检测框的初始特征与该检测框的初始特征对应的至少一个权重特征进行均值融合,即确定该检测框的初始特征与该检测框的初始特征对应的至少一个权重特征的平均值,然后电子设备将该平均值确定为该检测框的目标特征。
65.s103、电子设备基于每一个检测框的目标特征,对第一目标检测模型进行训练,以生成第二目标检测模型。
66.应理解,该第一目标检测模型为处于初始状态下的目标检测模型,该第二目标检测模型为已经训练完成的目标检测模型。
67.具体的,电子设备可以将该每一个检测框的目标特征输入该第一目标检测模型,得到该每一个检测框在该第一目标检测模型中的预测分类结果。然后,该电子设备可以结合该每一个检测框的真实分类结果等,确定该第一目标检测模型中的损失,并且基于该损失更新该第一目标检测模型中的参数,以生成该第二目标检测模型。
68.上述实施例提供的技术方案至少能够带来以下有益效果:由s101-s103可知,电子设备可以确定已检测图片包括的多个检测框中每一个检测框的初始特征以及该每一个检测框对应的至少一个权重特征,并且基于该每一个检测框的初始特征以及该每一个检测框对应的至少一个权重特征,生成该每一个检测框的目标特征。由于该每一个检测框对应的至少一个权重特征为第一分类器中包括的权重的特征,一个权重对应该第一分类器中的一个类别,而一个检测框对应的至少一个权重特征与该检测框之间的相似度大于或等于相似度阈值,因此,电子设备可以从该第一分类器包括的多个类别中,确定与该检测框的类别(具体为该检测框中包括的物体的类别)较为相似的至少一个类别(即该检测框的初始特征对应的至少一个权重特征各自对应的类别)。然后电子设备可以将该检测框的类别对应的特征(即该检测框的初始特征)与该至少一个类别对应的特征(即该检测框的初始特征对应的至少一个权重特征)相结合,可以实现从分类图片到检测图片上的类别的迁移,无需对目标检测模型的训练样本进行大量的人工标注,可以准确地生成能够表征该每一个检测框的类别(具体为该每一个检测框中包括的物体的类别)的真实数据分布的目标特征。然后,该电子设备可以基于该每一个检测框的目标特征对第一目标检测模型训练,以生成第二目标检测模型,能够提升目标检测模型的训练效率。进而,电子设备可以基于该第二目标检测模型快速、准确地预测出每一个待检测图片中包括的多个物体的类别,即能够提升目标检测模型的预测效率。
69.结合图1,如图2所示,在本公开实施例的一种实现方式中,上述第一目标检测模型中包括第一子网络模型和第二子网络模型,上述电子设备确定已检测图片包括的多个检测框中每一个检测框的初始特征,具体可以包括s1011-s1014。
70.s1011、电子设备获取已检测图片。
71.应理解,该已检测图片为上述第一目标检测模型的训练样本集中包括的训练样本,该已检测图片的数量可以为一个,也可以为多个。
72.s1012、电子设备将已检测图片输入第一子网络模型,进行特征识别,得到已检测图片的初始特征。
73.可选地,该第一子网络模型可以为一个卷积神经网络(convolutional neural networks,cnn),该cnn具体可以为resnet50、resnet101、resnet200、resnext101、resnest101、resnet18、mobilenet或者squeezenet等。
74.可选地,在电子设备将该已检测图片输入该第一子网络模型之前,还可以对该已检测图片进行裁剪处理。例如,裁剪处理后的已检测图片的短边长度可以为600像素,长边长度可以小于或等于1000像素。
75.s1013、电子设备将已检测图片的初始特征输入第二子网络模型,得到多个检测框中每一个检测框的位置以及每一个检测框的大小。
76.在本公开实施例的一种实现方式中,对于该多个检测框中的某一个检测框而言,可以以该检测框的中心点的坐标表征该检测框的位置。并且,可以以该检测框的长和宽表征该检测框的大小。
77.可选地,该第二子网络模型可以为区域提议网络(region proposal network,rpn)。
78.s1014、电子设备基于每一个检测框的位置以及每一个检测框的大小,从已检测图片的初始特征中提取每一个检测框的初始特征。
79.应理解,该已检测图片的初始特征中包括该已检测图片中包括的全部内容的初始特征。本公开实施例中,电子设备在得到多个检测框中每一个检测框的位置以及该每一个检测框的大小之后,可以基于某一个检测框的位置以及该检测框的大小确定该检测框在该已检测图片中的位置和大小,然后从该已检测图片的初始特征中提取该检测框在该已检测图片中的位置和大小对应的初始特征,即可以提取出该检测框的初始特征。
80.上述实施例提供的技术方案至少能够带来以下有益效果:由s1011-s1014可知,电子设备在获取到已检测图片之后,可以将已检测图片输入到第一目标检测模型中包括的第一子网络模型中,得到已检测图片的初始特征;并且将已检测图片的初始特征输入第一目标检测模型中包括的第二子网络模型,可以得到已检测图片包括的多个检测框中每一个检测框的位置以及该每一个检测框的大小;然后电子设备可以基于该每一个检测框的位置以及该每一个检测框的大小,从已检测图片的初始特征中提取该每一个检测框的初始特征。本公开实施例中,电子设备在可以基于已检测图片包括的某一个检测框的位置以及该检测框的大小确定该检测框在该已检测图片中的位置和大小,然后从该已检测图片的初始特征中提取该检测框在该已检测图片中的位置和大小对应的初始特征,即可以提取出该检测框的初始特征。能够准确、有效地提取出已检测图片中包括的每一个检测框的初始特征。
81.结合图2,如图3所示,在本公开实施例的一种实现方式中,上述第一分类器为上述第一图像分类模型中包括的分类器,该第一图片分类模型中包括第三子网络模型,该第三子网络模型中的参数与上述第一子网络模型中的参数相同,本公开实施例提供的目标检测模型的训练方法还可以包括s104-s106。
82.s104、电子设备获取多个分类图片。
83.应理解,该多个分类图片为该第一图片分类模型的训练样本集中包括的训练样本。
84.s105、电子设备将多个分类图片输入第三子网络模型,得到多个分类图片中每个分类图片的初始特征。
85.可以理解的是,该第三子网络模型中的参数与该第一子网络模型中的参数相同,说明该第三子网络模型与该第一子网络模型为相同的网络模型。该第三子网络模型与该第一子网络模型也可以理解为孪生网络。
86.可选地,该第三子网络模型也可以为一个cnn。
87.在一种可选的实现方式中,电子设备可以将上述已检测图片与该多个分类图片输入同一个网络模型,即上述第一图片分类模型中可以包括该第一子网络模型。电子设备在
获取都该多个分类图片之后,可以将该多个分类图片输入该第一子网络模型,以得到该多个分类图片中每个分类图片的初始特征。
88.可选地,在电子设备将该多个分类图片输入该第三子网络模型之前,也可以对该多个分类图片进行裁剪处理。例如,电子设备可以将该多个分类图片中的每个分类图片裁剪为224像素*224像素大小的图片。
89.需要说明的是,电子设备对上述已检测图片以及该多个分类图片进行裁剪处理,以保证同一批次内的训练样本可以同时进行训练。
90.可选地,为了提升模型训练的速度和准确度,电子设备可以使用imagenet数据集对上述第一子网络模型以及第三子网络模型进行预训练。
91.s106、电子设备将每个分类图片的初始特征输入第一分类器,得到每个分类图片在第一分类器中的预测分类结果。
92.结合上述实施例的描述,应理解,上述多个分类图片中每个分类图片中可以包括一个物体,该分类图片的类别即为该分类图片中包括的物体的类别。本公开实施例中,电子设备得到的该每个分类图片在该第一分类器中的预测分类结果,即为该每个分类图片中包括的物体在该第一分类器中的预测分类结果。
93.上述实施例提供的技术方案至少能够带来以下有益效果:由s104-s106可知,电子设备可以获取多个分类图片,并且将该多个分类图片输入第三子网络模型(该第三子网络模型中的参数与第一目标检测模型中包括的第一子网络模型的参数相同),得到该多个分类图片中每个分类图片的初始特征;然后该电子设备将该每个分类图片的初始特征输入第一分类器(即第一图片分类模型中包括的分类器),得到该每个分类图片在该第一分类器的预测分类结果。本公开实施例中,电子设备可以通过孪生网络(即第一子网络模型和第三子网络模型)将多个分类图片和已检测图片进行联合训练,能够快速地确定出该多个分类图片和该已检测图片各自对应的预测分类结果。进而,电子设备可以基于该多个分类图片和该已检测图片各自对应的预测分类结果,确定第一图片分类模型以及第一目标检测模型中存在的损失,以更新该第一图片分类模型中以及第一目标检测模型中的参数,提升模型训练的效率。
94.结合图1,如图4所示,本公开实施例提供的目标检测模型的训练方法还包括s107。
95.s107、电子设备确定第一损失和第二损失。
96.其中,该第一损失用于表征多个分类图片中每个分类图片的真实分类结果与该每个分类图片在上述第一分类器中的预测分类结果之间的不一致程度,第二损失用于表征上述多个检测框中每一个检测框的空间不一致程度,其中,一个检测框的空间不一致程度包括该检测框的真实位置与该检测框在该第一目标检测模型中的预测位置之间的不一致程度,以及该检测框的真实大小与该检测框在第一目标检测模型中的预测大小之间的不一致程度。
97.应理解,电子设备在获取到该多个分类图片时,也可以获取到该多个分类图片的真实分类结果(也可以理解为该多个分类图片的真实标签),一个分类图片的真实分类结果用于表征该分类图片中包括的物体的真实类别。同理,电子设备在获取到上述已检测图片时,也可以获取到该已检测图片包括的多个检测框中每一个检测框的真实位置以及该每一个检测框的真实大小,具体为可以在该已检测图片中体现(或标注)出该每一个检测框的真
实位置以及该每一个检测框的真实大小。
98.结合上述实施例的描述,应理解,电子设备将该多个分类图片输入上述第三网络子模型,得到该多个分类图片中每个分类图片的初始特征,然后将该每个分类图片的初始特征输入该第一分类器,即可以得到该每个分类图片在该第一分类器中的预测分类结果,该每个分类图片在该第一分类器中的预测分类结果也可以理解为该每个分类图片在上述第一图片分类模型中的预测分类结果。
99.在一种可选的实现方式中,对于该多个分类图片中的某一个分类图片而言,电子设备可以将该分类图片的真实分类结果定义为1,并且确定该分类图片在该第一分类器中被预测为该真实分类结果的分数(例如0.5),则电子设备可以确定该分类图片对应的损失为该真实分类结果与该分数之间的差值(即1-0.5=0.5),进而电子设备可以将该多个分类图片中每个分类图片对应的损失之和,确定为上述第一损失。
100.在另一种可选的实现方式中,电子设备还可以采用交叉熵函数的方式确定上述第一损失。例如,电子设备可以确定上述分类图片对应的损失为-log0.5。
101.可以理解的是,该第一目标检测模型中包括上述第二子网络模型,对于该多个检测框中的某一个检测框而言,该检测框在该第一目标检测模型中的预测位置为上述电子设备将已检测图片的初始特征输入该第二子网络模型之后,得到的该检测框的位置。同理,该检测框在该第一目标检测模型中的预测大小为该电子设备将该已检测图片的初始特征输入该第二子网络模型之后,得到的该检测框的大小。
102.在本公开实施例的一种实现方式中,可以以坐标的形式表征上述某一个检测框在该第一目标检测模型中的预测位置以及该检测框的真实位置,并且以长和宽的形式表征该检测框在该第一目标检测模型中的预测大小以及该检测框的真实大小。电子设备可以确定该检测框对应的空间损失满足下述公式:
103.l1=|x
1-x2| |y
1-y2| |w
1-w2| |h
1-h2|
104.其中,l1表示该检测框对应的空间损失,x1表示该检测框在该第一目标检测模型中的预测位置的横坐标,y1表示该预测位置的纵坐标,w1表示该检测框在该第一目标检测模型中的预测大小中的长度,h1表示该预测大小中的宽度,x2表示该检测框的真实位置的横坐标,y2表示该真实位置的纵坐标,w2表示该检测框的真实大小中的长度,h2表示该真实大小中的宽度。
105.可选地,上述第一损失的权重可以为0.001,上述第二损失的权重可以为1。
106.继续如图4所示,上述电子设备基于每一个检测框的目标特征,对第一目标检测模型训练,以生成第二目标检测模型,具体可以包括s1031-s1034。
107.s1031、电子设备将每一个检测框的目标特征输入第二分类器,得到每一个检测框在第二分类器中的预测分类结果。
108.其中,该第二分类器为上述第一目标检测模型中包括的分类器。
109.本公开实施例中,该第二分类器中包括的类别的数量可以小于上述第一分类器中包括的类别的数量,例如该第二分类器可以为80类分类器。
110.s1032、电子设备确定第三损失和第四损失。
111.其中,该第三损失用于表征上述每一个检测框的真实分类结果与该每一个检测框在该第二分类器中的预测分类结果之间的不一致程度,该第四损失用于表征该每一个检测
框中是否包括物体的真实结果与该每一个检测框在该第二分类器中被预测为是否包括物体的结果之间的不一致程度。
112.结合上述实施例的描述,应理解,电子设备在获取到上述已检测图片时,也可以获取该已检测图片包括的多个检测框中每一个检测框的真实分类结果,即该每一个检测框中包括的物体(或目标)的真实类别。
113.需要说明的是,电子设备确定该第三损失的解释说明与电子设备确定上述第一损失中的描述是相同或类似的,此处不再赘述。
114.可以理解的是,电子设备在获取到上述已检测图片时,也可以获取到该每一个检测框中是否包括物体的真实结果,具体的,当某一个检测框中的真实分类结果为某一个物体(或目标)时,电子设备可以确定该检测框中包括物体。相反地,即当该检测框中的真实分类结果是背景(即不是物体)时,电子设备可以确定该检测框中不包括物体。
115.对于上述多个检测框中某一个检测框而言,该检测框在第二分类器中被预测为包括物体的结果为该检测框在该第二分类器中被预测为各个类别的概率之和,该检测框在该第二分类器中被预测为不包括物体的结果为该检测框在该第二分类器中被预测为背景的概率。
116.s1033、电子设备根据第一损失、第二损失、第三损失以及第四损失,确定目标损失。
117.在一种可选的实现方式中,电子设备可以将该第一损失、该第二损失、该第三损失以及该第四损失的和,确定为该目标损失。
118.s1034、电子设备基于目标损失,更新第一目标检测模型中的参数,以生成第二目标检测模型。
119.应理解,电子设备更新该第一目标检测模型中的参数,具体为更新上述第二分类器中的参数、第一子网络模型中的参数以及第二子网络模型中的参数。
120.在一种可选的实现方式中,电子设备可以基于目标损失,使用随机梯度下降法(stochastic gradient descent,sgd)确定预设梯度,然后基于该预设梯度更新该第一目标检测模型中的参数,得到该第二目标检测模型。
121.可选地,在该第一目标检测模型的训练过程中,学习率可以设置为0.002,该第二目标检测模型的整个训练过程可以共训练30000轮,其中,在第24000轮是可以将学习率下降10倍,即学习率更新为0.0002。
122.在本发明实施例的一种实现方式中,电子设备还可以基于该目标损失,更新上述第一图片分类模型中的参数,以生成第二图片分类模型,该第二图片分类模型可以理解为已经训练完成的图片分类模型。其中,电子设备更新该第一图片分类模型中的参数,具体为更新上述第一分类器中的参数以及第三子网络模型中的参数。
123.上述实施例提供的技术方案至少能够带来以下有益效果:由s107,s1031-s1034可知,电子设备可以确定第一损失、第二损失、第三损失以及第四损失,该第一损失用于表征多个分类图片中每个分类图片的真实分类结果与该每个分类图片在第一分类器中的预测分类结果之间的不一致程度,该第二损失用于表征已检测图片包括的多个检测框中每一个检测框的空间不一致程度(包括一个检测框的真实位置与该检测框在第一目标检测模型中的预测位置的不一致程度,以及该检测框的真实大小与该检测框在该目标检测模型中的预
测大小之间的不一致程度),该第三损失用于表征该每一个检测框的真实分类结果与该每一个检测框在第二分类器中的预测分类结果之间的不一致程度,该第四损失用于表征该每一个检测框中是否包括物体的真实结果与该每一个检测框在该第二分类器中被预测为是否包括物体的结果之间的不一致程度。然后该电子设备可以根据该第一损失、该第二损失、该第三损失以及该第四损失确定目标损失,并且基于该目标损失,更新第一目标检测模型中的参数,以生成第二目标检测模型。本公开实施例中,电子设备可以确定出第一目标检测模型以及第一图片分类模型在训练过程中存在的每一项损失,并且基于该每一项损失更新该第一目标检测模型以及该第一图片分类模型中的参数,能够准确、有效地对目标检测模型进行训练,以生成预测准确度较高的目标检测模型。
124.结合图1,如图5所示,本公开实施例提供的目标检测模型的训练方法还可以包括s108-s110。
125.s108、电子设备将第一检测框的目标特征确定为预设均值。
126.其中,该第一检测框为上述多个检测框的一个。
127.结合上述实施例的描述,应理解,电子设备可以基于该多个检测框中每一个检测框的初始特征以及该每一个检测框的初始特征对应的至少一个权重,生成该每一个检测框的目标特征。然后,对于该多个检测框中的某一个检测框(例如该第一检测框)而言,电子设备可以将该第一检测框的目标特征确定为上述预设均值。
128.s109、电子设备基于预设均值和预设标准差,生成第一检测框的多个拓展特征。
129.在一种可选的实现方式中,电子设备可以基于该预设均值和该预设标准差,并且采用高斯函数生成该第一检测框的多个扩展特征。
130.具体的,电子设备可以确定该第一检测框的多个扩展特征满足下述公式:
131.ig=gaussian(i
t
,σ)
132.其中,ig表示该第一检测框的多个扩展特征,i
t
表示该预设均值,σ表示该预设标准差,gaussian表示高斯函数。
133.示例性的,电子设备可以将该预设标准差设置为1。
134.s110、电子设备基于多个检测框中每一个检测框的多个扩展特征,对第一目标检测模型进行训练。
135.应理解,电子设备可以基于上述预设均值和预设标准差,生成更多能够表征该多个检测框中每一个检测框的类别的特征(即该每一个检测框的多个扩展特征),能够提升待训练特征(即用于对第一目标检测模型进行训练的特征)的丰富性。进而,电子设备基于该每个检测框的多个扩展特征对该第一目标检测模型进行训练时,可以训练出预测准确度较高的目标检测模型。
136.需要说明的是,电子设备基于该多个检测框中每一个检测框的多个扩展特征,对该第一目标检测模型进行训练的解释说明,与上述电子设备基于该每一个检测框的目标特征,对该第一目标检测模型进行训练中的描述是相同或类似的,此处不再赘述。
137.上述实施例提供的技术方案至少能够带来以下有益效果:由s108-s110可知,电子设备可以将多个检测框中的某一个检测框(例如第一检测框)的目标特征确定为预设均值,并且基于该预设均值和预设标准差,生成该第一检测框的多个扩展特征。由于该多个检测框中每一个检测框的多个扩展特征能够提升模型训练时的丰富性,如此,电子设备在基于
该每一个检测框的多个拓展特征,对第一目标检测模型进行训练时,可以训练出预测准确度较高的目标检测模型。
138.如图6所示,本公开实施例提供的目标检测方法可以包括s201-s202。
139.s201、电子设备获取待检测图片。
140.应理解,该待检测图片为一个没有真实分类结果的检测图片。
141.s202、电子设备将待检测图片输入第二目标检测模型,得到待检测图片包括的至少一个检测框各自的类别。
142.其中,该第二目标检测模型是基于上述本公开实施例中提供的目标检测模型的训练方法训练得到的。
143.具体的,该第二目标检测模型是基于已检测图片包括的多个检测框中每一个检测框的目标特征对第一目标检测模型训练生成的,该每一个检测框的目标特征是基于该每一个检测框的初始特征以及该每一个检测框的初始特征对应的至少一个权重特征生成的,其中,一个检测框的初始特征对应的至少一个权重特征中每一个权重特征与该检测框的初始特征之间的相似度大于或等于相似度阈值,该每一个检测框的初始特征对应的至少一个权重特征为第一分类器中包括的权重的特征,一个权重对应该第一分类器中的一个类别,该第一类别为第一图片分类模型中包括的分类器。
144.结合上述实施例的描述,应理解,该第一目标检测模型为初始状态下的目标检测模型,该第二目标检测模型为已经训练完成的目标检测模型,该第一图片分类模型为初始状态下的图片分类模型,该待检测图片包括的至少一个检测框各自的类别为该至少一个检测框各自包括的物体(或目标)的类别。
145.可以理解的是,上述第二图片分类模型为已经训练完成的图片分类模型。电子设备可以获取某一待识别分类图片,并且将该待识别分类图片输入该第二图片分类模型,以得到该待识别分类图片(具体为该待识别分类图片中包括的物体)的类别。
146.需要说明的是,电子设备生成该多个检测框中每一个检测框的目标特征的具体过程,以及电子设备基于该每一个检测框的目标特征对该第一目标检测模型训练,以生成该第二目标检测模型的具体过程可以参见上述实施例的描述,此处不再赘述。
147.上述实施例提供的技术方案至少能够带来以下有益效果:由s201-s202可知,电子设备可以获取待检测图片,并且将待检测图片输入第二目标检测模型,得到待检测图片包括的至少一个检测框各自的类别。本公开实施例中,由于一个检测框的初始特征与检测框的初始特征对应的至少一个权重特征之间的相似度大于或等于相似度阈值,说明电子设备可以从第一分类器中包括的多个类别中确定与该检测框的类别较为相似的至少一个类别(即该检测框的初始特征对应的至少一个权重特征各自对应的类别)。该检测框的目标特征可以实现从分类图片到检测图片上的类别的迁移,无需对目标检测模型的训练样本进行大量的人工标注,能够准确地表征出每一个检测框的类别,即该第二目标检测模型为预测速度高,并且预测准确度高的目标检测模型。如此,电子设备可以基于该第二目标检测模型准确地确定出该待检测图片包括的至少一个检测框各自的类别,能够提升目标检测的效率。
148.结合图6,如图7所示,上述第二目标检测模型中包括第一子网络模型,第二子网络模型以及第二分类器,上述电子设备将待检测图片输入第二目标检测模型,得到待检测图片包括的至少一个检测框各自的类别,具体可以包括s2021-s2024。
149.s2021、电子设备将待检测图片输入第一子网络模型,进行特征识别,得到待检测图片的初始特征。
150.s2022、电子设备将待检测图片的初始特征输入第二子网络模型,得到至少一个检测框各自的位置以及至少一个检测框各自的大小。
151.s2023、电子设备基于至少一个检测框各自的位置以及至少一个检测框各自的大小,从待检测图片的初始特征中提取至少一个检测框各自的初始特征。
152.s2024、电子设备将至少一个检测框各自的初始特征输入第二分类器,得到至少一个检测框各自的类别。
153.需要说明的是,上述s2021-s2024中的解释说明可以参见上述实施例中的描述,此处不再赘述。
154.上述实施例提供的技术方案至少能够带来以下有益效果:由s2021-s2024可知,电子设备可以将待检测图片输入到第二目标检测模型中包括的第一子网络模型中,得到该待检测图片的初始特征;并且将该待检测图片的初始特征输入该第二目标检测模型中包括的第二子网络模型,可以得到该待检测图片包括的至少一个检测框各自的位置以及该至少一个检测框各自的大小;然后该电子设备可以基于该至少一个检测框各自的位置以及该至少一个检测框各自的大小从该待检测图片的初始特征中提取该至少一个检测框各自的初始特征,以及将该至少一个检测框各自的初始特征输入该第二目标检测模型中包括的第二分类器中,以得到该至少一个检测框各自的类别。本公开实施例中,电子设备能够准确、有效地确定出待检测图片中包括的至少一个检测框各自的类别,即可以准确、有效地确定出该至少一个检测框各自包括的物体的类别。
155.可以理解的,在实际实施时,本公开实施例所述的电子设备可以包含有用于实现前述对应目标检测模型的训练方法以及目标检测方法的一个或多个硬件结构和/或软件模块,这些执行硬件结构和/或软件模块可以构成一个电子设备。本领域技术人员应该很容易意识到,结合本文中所公开的实施例描述的各示例的算法步骤,本公开能够以硬件或硬件和计算机软件的结合形式来实现。某个功能究竟以硬件还是计算机软件驱动硬件的方式来执行,取决于技术方案的特定应用和设计约束条件。专业技术人员可以对每个特定的应用来使用不同方法来实现所描述的功能,但是这种实现不应认为超出本公开的范围。
156.基于这样的理解,本公开实施例还对应提供一种目标检测模型的训练装置,图8示出了本公开实施例提供的目标检测模型的训练装置的结构示意图。如图8所示,该目标检测模型的训练装置10可以包括:确定模块101和处理模块102。
157.确定模块101,被配置为确定已检测图片包括的多个检测框中每一个检测框的初始特征以及该每一个检测框的初始特征对应的至少一个权重特征,其中,一个检测框的初始特征对应的至少一个权重特征中每一个权重特征与该检测框的初始特征之间的相似度大于或等于相似度阈值,该每一个检测框的初始特征对应的至少一个权重特征为第一分类器中包括的权重的特征,一个权重对应该第一分类器中的一个类别。
158.处理模块102,被配置为基于该多个检测框中每一个检测框的初始特征以及该每一个检测框的初始特征对应的至少一个权重特征,生成该每一个检测框的目标特征。
159.处理模块102,还被配置为基于该每一个检测框的目标特征,对第一目标检测模型训练,以生成第二目标检测模型。
160.可选地,上述第一目标检测模型中包括第一子网络模型和第二子网络模型,上述目标检测模型的训练装置10还包括获取模块103。
161.获取模块103,被配置为获取该已检测图片。
162.处理模块102,具体被配置为将该已检测图片输入该第一子网络模型,进行特征识别,得到该已检测图片的初始特征。
163.处理模块102,具体还被配置为将该已检测图片的初始特征输入该第二子网络模型,得到该多个检测框中每一个检测框的位置以及该每一个检测框的大小。
164.处理模块102,具体还被配置为基于该每一个检测框的位置以及该每一个检测框的大小,从该已检测图片的初始特征中提取该每一个检测框的初始特征。
165.可选地,上述第一分类器为第一图片分类模型中包括的分类器,该第一图片分类模型中包括第三子网络模型,该第三子网络模型中的参数与该第一子网络模型中的参数相同。
166.获取模块103,还被配置为获取多个分类图片。
167.处理模块102,还被配置为将该多个分类图片输入该第三子网络模型,得到该多个分类图片中每个分类图片的初始特征;
168.处理模块102,还被配置为将该每个分类图片的初始特征输入该第一分类器,得到该每个分类图片在该第一分类器中的预测分类结果。
169.可选地,确定模块101,还被配置为确定第一损失和第二损失,该第一损失用于表征多个分类图片中每个分类图片的真实分类结果与该每个分类图片在该第一分类器中的预测分类结果之间的不一致程度,该第二损失用于表征该多个检测框中每一个检测框的空间不一致程度,其中,一个检测框的空间不一致程度包括该检测框的真实位置与该检测框在该第一目标检测模型中的预测位置之间的不一致程度,以及该检测框的真实大小与该检测框在该第一目标检测模型中的预测大小之间的不一致程度。
170.处理模块102,具体被配置为将该每一个检测框的目标特征输入第二分类器,得到该每一个检测框在该第二分类器中的预测分类结果,该第二分类器为该第一目标检测模型中包括的分类器。
171.确定模块101,具体被配置为确定第三损失和第四损失,该第三损失用于表征该每一个检测框的真实分类结果与该每一个检测框在该第二分类器中的预测分类结果之间的不一致程度,该第四损失用于表征该每一个检测框中是否包括物体的真实结果与该每一个检测框在该第二分类器中被预测为是否包括物体的结果之间的不一致程度。
172.确定模块101,具体还被配置为根据该第一损失、该第二损失、该第三损失以及该第四损失,确定目标损失。
173.处理模块102,具体还被配置为基于该目标损失,更新该第一目标检测模型中的参数,以生成该第二目标检测模型。
174.可选地,确定模块101,还被配置为将第一检测框的目标特征确定为预设均值,该第一检测框为该多个检测框中的一个。
175.处理模块102,还被配置为基于该预设均值和预设标准差,生成该第一检测框的多个拓展特征。
176.处理模块102,还被配置为基于该多个检测框中每一个检测框的多个拓展特征,对
该第一目标检测模型进行训练。
177.如上所述,本公开实施例可以根据上述方法示例对目标检测模型的训练装置进行功能模块的划分。其中,上述集成的模块既可以采用硬件的形式实现,也可以采用软件功能模块的形式实现。另外,还需要说明的是,本公开实施例中对模块的划分是示意性的,仅仅为一种逻辑功能划分,实际实现时可以有另外的划分方式。例如,可以对应各个功能划分各个功能模块,也可以将两个或两个以上的功能集成在一个处理模块中。
178.关于上述实施例中的目标检测模型的训练装置,其中各个模块执行操作的具体方式、以及具备的有益效果,均已经在前述方法实施例中进行了详细描述,此处不再赘述。
179.图9是本公开提供的另一种目标检测模型的训练装置的结构示意图。如图9,该目标检测模型的训练装置20可以包括至少一个处理器201以及用于存储处理器可执行指令的存储器203。其中,处理器201被配置为执行存储器203中的指令,以实现上述实施例中的目标检测模型的训练方法。
180.另外,目标检测模型的训练装置20还可以包括通信总线202以及至少一个通信接口204。
181.处理器201可以是一个处理器(central processing units,cpu),微处理单元,asic,或一个或多个用于控制本公开方案程序执行的集成电路。
182.通信总线202可包括一通路,在上述组件之间传送信息。
183.通信接口204,使用任何收发器一类的装置,用于与其他设备或通信网络通信,如以太网,无线接入网(radio access network,ran),无线局域网(wireless local area networks,wlan)等。
184.存储器203可以是只读存储器(read-only memory,rom)或可存储静态信息和指令的其他类型的静态存储设备,随机存取存储器(random access memory,ram)或者可存储信息和指令的其他类型的动态存储设备,也可以是电可擦可编程只读存储器(electrically erasable programmable read-only memory,eeprom)、只读光盘(compact disc read-only memory,cd-rom)或其他光盘存储、光碟存储(包括压缩光碟、激光碟、光碟、数字通用光碟、蓝光光碟等)、磁盘存储介质或者其他磁存储设备、或者能够用于携带或存储具有指令或数据结构形式的期望的程序代码并能够由计算机存取的任何其他介质,但不限于此。存储器可以是独立存在,通过总线与处理单元相连接。存储器也可以和处理单元集成在一起。
185.其中,存储器203用于存储执行本公开方案的指令,并由处理器201来控制执行。处理器201用于执行存储器203中存储的指令,从而实现本公开方法中的功能。
186.在具体实现中,作为一种实施例,处理器201可以包括一个或多个cpu,例如图9中的cpu0和cpu1。
187.在具体实现中,作为一种实施例,目标检测模型的训练装置20可以包括多个处理器,例如图9中的处理器201和处理器207。这些处理器中的每一个可以是一个单核(single-cpu)处理器,也可以是一个多核(multi-cpu)处理器。这里的处理器可以指一个或多个设备、电路、和/或用于处理数据(例如计算机程序指令)的处理核。
188.在具体实现中,作为一种实施例,目标检测模型的训练装置20还可以包括输出设备205和输入设备206。输出设备205和处理器201通信,可以以多种方式来显示信息。例如,
输出设备205可以是液晶显示器(liquid crystal display,lcd),发光二级管(light emitting diode,led)显示设备,阴极射线管(cathode ray tube,crt)显示设备,或投影仪(projector)等。输入设备206和处理器201通信,可以以多种方式接受用户的输入。例如,输入设备206可以是鼠标、键盘、触摸屏设备或传感设备等。
189.图10为本公开提供的一种目标检测装置的结构示例图。如图10所示,该目标检测装置30可以包括:根据本公开实施例的第四方面,提供一种目标检测装置。该目标检测装置30可以包括:获取模块301和处理模块302。
190.获取模块301,被配置为获取待检测图片。
191.处理模块302,被配置为将该待检测图片输入第二目标检测模型,得到该待检测图片包括的至少一个检测框各自的类别,该第二目标检测模型是基于上述本公开实施例中提供的目标检测模型的训练方法训练得到的。
192.可选地,第二目标检测模型中包括第一子网络模型、第二子网络模型以及第二分类器。
193.处理模块302,具体被配置为将该待检测图片输入该第一子网络模型,进行特征识别,得到该待检测图片的初始特征。
194.处理模块302,具体还被配置为将该待检测图片的初始特征输入该第二子网络模型,得到该至少一个检测框各自的位置以及该至少一个检测框各自的大小。
195.处理模块302,具体还被配置为基于该至少一个检测框各自的位置以及该至少一个检测框各自的大小,从该待检测图片的初始特征中提取该至少一个检测框各自的初始特征。
196.处理模块302,具体还被配置为将该至少一个检测框各自的初始特征输入该第二分类器,得到该至少一个检测框各自的类别。
197.图11是本公开提供的另一种目标检测装置的结构示意图,该目标检测装置可以是上述电子设备。如图11,该目标检测装置40可以包括至少一个处理器401以及用于存储处理器可执行指令的存储器403。其中,处理器401被配置为执行存储器403中的指令,以实现上述实施例中的目标检测方法。
198.另外,目标检测装置40还可以包括通信总线402以及至少一个通信接口404。
199.处理器401可以是一个cpu,微处理单元,asic,或一个或多个用于控制本公开方案程序执行的集成电路。
200.通信总线402可包括一通路,在上述组件之间传送信息。
201.通信接口404,使用任何收发器一类的装置,用于与其他设备或通信网络通信,如以太网,ran,wlan等。
202.存储器403可以是rom或可存储静态信息和指令的其他类型的静态存储设备,ram或者可存储信息和指令的其他类型的动态存储设备,也可以是eeprom、cd-rom或其他光盘存储、光碟存储(包括压缩光碟、激光碟、光碟、数字通用光碟、蓝光光碟等)、磁盘存储介质或者其他磁存储设备、或者能够用于携带或存储具有指令或数据结构形式的期望的程序代码并能够由计算机存取的任何其他介质,但不限于此。存储器可以是独立存在,通过总线与处理单元相连接。存储器也可以和处理单元集成在一起。
203.其中,存储器403用于存储执行本公开方案的指令,并由处理器401来控制执行。处
理器401用于执行存储器403中存储的指令,从而实现本公开方法中的功能。
204.在具体实现中,作为一种实施例,处理器401可以包括一个或多个cpu,例如图11中的cpu0和cpu1。
205.在具体实现中,作为一种实施例,目标检测装置40可以包括多个处理器,例如图11中的处理器401和处理器407。这些处理器中的每一个可以是一个单核(single-cpu)处理器,也可以是一个多核(multi-cpu)处理器。这里的处理器可以指一个或多个设备、电路、和/或用于处理数据(例如计算机程序指令)的处理核。
206.在具体实现中,作为一种实施例,目标检测装置40还可以包括输出设备405和输入设备406。输出设备405和处理器401通信,可以以多种方式来显示信息。例如,输出设备405可以是lcd,led显示设备,crt显示设备,或投影仪(projector)等。输入设备406和处理器401通信,可以以多种方式接受用户的输入。例如,输入设备406可以是鼠标、键盘、触摸屏设备或传感设备等。
207.本领域技术人员可以理解,图9中示出的结构并不构成对目标检测模型的训练装置20的限定,并且图11中示出的结构并不构成对目标检测装置40的限定。其可以包括比图示更多或更少的组件,或者组合某些组件,或者采用不同的组件布置。
208.另外,本公开还提供一种计算机可读存储介质,包括指令,当指令由电子设备执行时,使得电子设备执行如上述实施例所提供的目标检测模型的训练方法,或者执行如上述实施例所提供的目标检测方法。
209.另外,本公开还提供一种计算机程序产品,包括指令,当指令由电子设备执行时,使得电子设备执行如上述实施例所提供的目标检测模型的训练方法,或者执行如上述实施例所提供的目标检测方法。
210.本领域技术人员在考虑说明书及实践这里公开的发明后,将容易想到本公开的其它实施方案。本公开旨在涵盖本公开的任何变型、用途或者适应性变化,这些变型、用途或者适应性变化遵循本公开的一般性原理并包括本公开未公开的本技术领域中的公知常识或惯用技术手段。说明书和实施例仅被视为示例性的,本公开的真正范围和精神由权利要求指出。
再多了解一些

本文用于企业家、创业者技术爱好者查询,结果仅供参考。

发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表

相关文献