[发明专利]一种关键点检测模型训练方法、关键点检测方法和装置

摘要

本申请实施例公开一种关键点检测模型训练方法、关键点检测方法和装置，在关键点检测模型训练中同时利用有标注和无标注图像样本。在训练时，通过生成模型根据获取的未标注图像样本生成第一热力图，根据获取的标注图像样本中标注的关键点坐标确定第二热力图。通过判别模型计算第一热力图和未标注图像样本的第一匹配度，以及计算第二热力图和标注图像样本的第二匹配度，根据第一匹配度和第二匹配度构建对抗损失函数。从而根据该对抗损失函数对生成模型和判别模型进行训练。由于在训练时采用了大量无标注图像样本，仅需要少量的标注图像样本，降低了人工标注成本，提高了模型训练的效率。同时，减小了人工标注带来的标注偏差，提升了模型训练的效果。