人脸关键点定位模型的训练方法、装置、设备及存储介质
摘要:
本申请实施例公开了一种人脸关键点定位模型的训练方法、装置、设备及存储介质。所述方法包括:构建用于人脸关键点定位的CNN模型,CNN模型的卷积层数量大于第一阈值,且卷积层的通道数小于第二阈值;采用CNN模型对训练样本进行人脸关键点定位,得到人脸关键点的预测位置;其中,人脸关键点包括n种分类,n为大于1的整数;分别每一种分类的人脸关键点的预测位置和真实位置,计算n种分类各自对应的损失函数值,进而计算CNN模型的损失函数值;当CNN模型的损失函数值小于预设阈值时,停止对CNN模型的训练并保存。本申请实施例通过构建瘦长型的CNN模型,在尽可能保证定位精度不受损失的前提下,减小模型体积。
0/0