关键点检测方法、回归模型的训练方法、装置及电子设备
摘要:
本申请实施例公开了一种关键点检测方法、回归模型的训练方法、装置及电子设备,通过基于回归模型对样本图像进行特征映射,输出样本图像所承载的对象的样本关键点坐标,以及样本关键点坐标对应的样本热图分布参数;将样本关键点坐标作为均值,根据样本热图分布参数生成样本关键点坐标对应的模拟热图;获取样本图像对应的参考热图,根据模拟热图和参考热图确定回归模型的目标损失,根据目标损失训练回归模型;然后,获取待检测的目标图像,基于训练后的回归模型对目标图像进行特征映射,输出目标图像所承载的对象的目标关键点坐标,能够有效地提升关键点检测的准确性,可广泛应用于人工智能、智慧交通、辅助驾驶等多种场景。
0/0