【技术实现步骤摘要】
特征检测方法、模型合并方法、设备和介质
[0001]本专利技术涉及计算机视觉
,尤其是涉及特征检测方法、模型合并方法、设备和介质。
技术介绍
[0002]由于实际业务的变动,越来越多的业务更倾向于边缘部署。边缘部署主要用于嵌入式设备,主要通过将模型打包封装到软件开发工具包(Software Development Kit,SDK),集成到嵌入式设备,数据的处理和模型推理都在终端设备上执行。但是现有的模型个数太多,若一一进行部署,将导致边缘部署十分不利。
技术实现思路
[0003]基于此,有必要针对上述问题,提供特征检测方法、模型合并方法、设备和介质,以解决现有模型不利于边缘部署的问题。
[0004]一种模型合并方法,应用于模型集合,所述模型集合包括网络结构相同的多个教师模型和一个学生模型,所述多个教师模型均已收敛,不同教师模型用于检测不同特征,所述学生模型未收敛,所述方法包括:
[0005]获取多个未携带标签的人脸训练图像,分别将第一人脸图像输入所述多个教师模型,获取每个教师模型输出的第 ...
【技术保护点】
【技术特征摘要】
1.一种模型合并方法,其特征在于,应用于模型集合,所述模型集合包括网络结构相同的多个教师模型和一个学生模型,所述多个教师模型均已收敛,不同教师模型用于检测不同特征,所述学生模型未收敛,所述方法包括:获取多个未携带标签的人脸训练图像,分别将第一人脸图像输入所述多个教师模型,获取每个教师模型输出的第一特征信息,将所有教师模型输出的第一特征信息作为所述第一人脸图像的标签;其中,所述第一人脸图像为所述多个未携带标签的人脸训练图像中的任意一个,不同教师模型输出的第一特征信息不同;将携带有标签的第一人脸图像输入所述学生模型,获取所述学生模型输出的对应多个特征的第二特征信息;通过预设的蒸馏损失函数,根据所述第一特征信息和所述第二特征信息计算蒸馏损失,根据所述蒸馏损失调整所述学生模型的参数,返回执行所述分别将第一人脸图像输入所述多个教师模型的步骤,直至所述学生模型收敛。2.根据权利要求1所述的方法,其特征在于,所述第一特征信息和所述第二特征信息均包括特征严重程度及置信度,所述蒸馏损失函数包括多个特征蒸馏损失函数,一个特征蒸馏损失函数对应一个特征,一个特征蒸馏损失函数包括一个软标签损失函数及一个硬标签损失函数;所述通过预设的蒸馏损失函数,根据所述第一特征信息和所述第二特征信息计算蒸馏损失,包括:通过所述软标签损失函数,根据所述第一特征信息中目标特征对应的特征严重程度和所述第二特征信息中目标特征对应的特征严重程度计算所述目标特征的软标签损失;其中,所述目标特征为所述多个教师模型检测的特征中的任意一个;通过所述硬标签损失函数,根据所述第一特征信息中目标特征对应的置信度与所述第二特征信息中目标特征对应的特征严重程度计算所述目标特征的硬标签损失;计算第一权重与所述软标签损失的第一乘积,及第二权重与所述硬标签损失的第二乘积,将所述第一乘积与所述第二乘积的和作为所述目标特征对应的特征蒸馏损失;其中,所述第一权重与所述第二权重的和为1;计算所有特征对应的特征蒸馏损失的和,以得到所述蒸馏损失。3.根据权利要求2所述的方法,其特征在于,所述软标签损失函数的公式为:其中,N为目标特征对应的特征严重程度的总程度数;为所述第一特征信息中目标特征对应的特征严重程度,指示在温度=T的情况下,第一特征信息中目标特征对应的特征严重程度在程度j上的值;为所述第二特征信息中目标特征对应的特征严重程度,指示在温度=T的情况下,第二特征信息中目标特征对应的特征严重程度在程度j上的值。4.根据权利要求3所述的方法,其特征在于,
其中,v
j
指示教师模型中全连接层上目标特征在程度j的输出;指...
【专利技术属性】
技术研发人员:曾梦萍,
申请(专利权)人:深圳数联天下智能科技有限公司,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。