模型训练方法、装置、电子设备和计算机可读存储介质
摘要:
本申请实施例涉及深度学习技术领域,公开了一种模型训练方法、装置、电子设备和计算机可读存储介质。上述模型训练方法包括:根据训练样本预标注的关键点的坐标,获取训练样本的人脸姿态角粗标签;其中,训练样本为二维人脸图像,关键点包括左眼睛中心、右眼睛中心、鼻尖、左嘴角和右嘴角,人脸姿态角粗标签包括俯仰角和偏航角;根据人脸姿态角粗标签和预设的候选姿态角集合,获取训练样本的人脸姿态角的联合概率分布;根据训练样本、联合概率分布和预设的损失函数,训练预设的深度学习网络,得到人脸姿态角估计模型。本申请提供的模型训练方法,训练过程的计算量很小,训练所需的时间比较短,可以快速获得稳定、可靠的人脸姿态角估计模型。
0/0