关键点检测及检测网络训练的方法、电子设备和存储介质
摘要:
本发明实施例涉及神经网络领域,公开了一种关键点检测及检测网络训练的方法、电子设备和存储介质。本发明中,获取预设的关键点模板、训练样本图像和所述训练样本图像对应的样本关键点,所述关键点模板为预先构建的初始关键点检测网络输出结果的形状标准;通过所述初始关键点检测网络,结合所述训练样本图像和所述关键点模板获得训练检测关键点;根据所述训练检测关键点和所述样本关键点对所述初始关键点检测网络进行迭代训练,并将满足预设迭代条件的所述初始关键点检测网络作为训练完成的关键点检测网络。提供了关键点模板作为先验,保证关键点形状正常,提高关键点检测的准确性。
0/0