【技术实现步骤摘要】
一种人脸比对的图卷积神经网络训练方法及装置
本申请实施例涉及图像处理
,尤其涉及一种人脸比对的图卷积神经网络训练方法及装置。
技术介绍
随着深度学习技术的快速发展,人脸识别的精度得到了显著的提升,人脸识别技术在商业和安防场景也得到大量应用,其中对未打标签的人脸图片进行人脸聚类是一种比较常见的应用。在进行人脸聚类过程中,两组人脸图片进行m:n比对合并判断是人脸聚类场景中比较常见的问题。现有技术中经常利用卷积神经网络进行人脸聚类,在卷积神经网络投入使用之前,需要对卷积神经网络进行训练,但是对卷积神经网络的训练一般会基于大量的人脸图片进行,导致训练效率较低。
技术实现思路
本申请实施例提供一种人脸比对的图卷积神经网络训练方法及装置,以提升对人脸识别图卷积神经网络的训练效率。在第一方面,本申请实施例提供了一种人脸比对的图卷积神经网络训练方法,包括:基于设定的邻域半径,根据每个人脸图片之间的相似度距离为每个所述人脸图片构建人脸图片邻域;按照设定的相似度分割阈值,根据所述人脸图片邻域构造同类低相似度的正样本集和不同类高相似度的负样本集;基于图卷积神经网络构建二分类的多对多人脸识别模型;依据所述正样本集和所述负样本集对所述多对多人脸识别模型进行训练,以使所述多对多人脸识别模型的正确率达到设定要求。进一步的,所述基于设定的邻域半径,根据每个人脸图片之间的相似度距离为每个所述人脸图片构建人脸图片邻域,包括:基于人脸图片数据集,确定两两人脸图片之间 ...
【技术保护点】
1.一种人脸比对的图卷积神经网络训练方法,其特征在于,包括:/n基于设定的邻域半径,根据每个人脸图片之间的相似度距离为每个所述人脸图片构建人脸图片邻域;/n按照设定的相似度分割阈值,根据所述人脸图片邻域构造同类低相似度的正样本集和不同类高相似度的负样本集;/n基于图卷积神经网络构建二分类的多对多人脸识别模型;/n依据所述正样本集和所述负样本集对所述多对多人脸识别模型进行训练,以使所述多对多人脸识别模型的正确率达到设定要求。/n
【技术特征摘要】
1.一种人脸比对的图卷积神经网络训练方法,其特征在于,包括:
基于设定的邻域半径,根据每个人脸图片之间的相似度距离为每个所述人脸图片构建人脸图片邻域;
按照设定的相似度分割阈值,根据所述人脸图片邻域构造同类低相似度的正样本集和不同类高相似度的负样本集;
基于图卷积神经网络构建二分类的多对多人脸识别模型;
依据所述正样本集和所述负样本集对所述多对多人脸识别模型进行训练,以使所述多对多人脸识别模型的正确率达到设定要求。
2.根据权利要求1所述的人脸比对的图卷积神经网络训练方法,其特征在于,所述基于设定的邻域半径,根据每个人脸图片之间的相似度距离为每个所述人脸图片构建人脸图片邻域,包括:
基于人脸图片数据集,确定两两人脸图片之间的相似度距离,所述人脸图片数据集保存有多个确定人脸分类的人脸图片;
对于每个所述人脸图片,确定在同一人脸分类中,对应所述相似度距离在设定的邻域半径以内的其他所述人脸图片;
基于确定的所述人脸图片为每个所述人脸图片构建人脸图片邻域。
3.根据权利要求1所述的人脸比对的图卷积神经网络训练方法,其特征在于,所述按照设定的相似度分割阈值,根据所述人脸图片邻域构造同类低相似度的正样本集和不同类高相似度的负样本集,包括:
分别确定每个人脸分类中,两两所述人脸图片之间的相似度距离,并确定对应相似度距离大于设定的相似度分割阈值的正人脸图片组;
确定所述正人脸图片组中每张所述人脸图片对应的人脸图片邻域,并基于所述人脸图片邻域构造同类低相似度的正样本集;
分别确定不同人脸分类中,所述人脸图片之间的相似度距离,并确定对应相似度距离小于等于设定的相似度分割阈值的负人脸图片组;
确定所述负人脸图片组中每张所述人脸图片对应的人脸图片邻域,并基于所述人脸图片邻域构造不同类高相似度的负样本集。
4.根据权利要求1所述的人脸比对的图卷积神经网络训练方法,其特征在于,所述多对多人脸识别模型为:
Hl=δ(D-1AHl-1Wl-1)
Y=g(HL)
其中,Hl为卷积神经网络的第l层隐藏向量,是一个(m+n)×dl矩阵,m和n分别为两组人脸图片邻域包含的人脸图片的图片数量,dl表示第l层的维度,l的取值为1~L,L为设定的最后一个隐藏层,Wl-1为模型第l层的训练参数,A为两组人脸图片的相似度距离矩阵,D为相似度距离矩阵的度矩阵,A和D是(m+n)×(m+n)矩阵,δ为神经网络激活函数...
【专利技术属性】
技术研发人员:李逸帆,秦伟,丁保剑,
申请(专利权)人:佳都新太科技股份有限公司,广州佳都科技软件开发有限公司,
类型:发明
国别省市:广东;44
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。