[发明专利]热力图转换模型训练方法以及装置

权利要求书

1.一种热力图转换模型训练方法，包括：

对目标对象的特征图像进行特征点标注，并生成所述特征图像对应的至少一个特征点坐标；

根据目标特征点坐标确定所述目标特征点坐标对应的热力区域，并基于所述热力区域生成所述目标特征点对应的第一特征点热力图，其中，所述目标特征点坐标为所述至少一个特征点坐标之一；

将所述特征图像作为样本图像，并将所述第一特征点热力图作为样本标签输入待训练热力图转换模型；

获取所述待训练热力图转换模型输出的所述特征图像对应的第二特征点热力图；

判断所述第二特征点热力图的尺寸与所述第一特征点热力图的尺寸是否一致；

若否，则根据所述第一特征点热力图的尺寸，对所述第二特征点热力图进行尺寸变换，生成变换后的第二特征点热力图；

计算变换后的第二特征点热力图与所述第一特征点热力图之间的损失值；

根据所述损失值对所述待训练热力图转换模型进行迭代训练，生成热力图转换模型，所述热力图转换模型使得所述样本标签与所述特征图像相关联。

2.根据权利要求1所述的热力图转换模型训练方法，还包括：

根据所述第一特征点热力图中各像素点的坐标与所述第二特征点热力图中各像素点的坐标间的距离计算所述损失值。

3.根据权利要求1所述的热力图转换模型训练方法，还包括：

确定所述第二特征点热力图中分别包含的热力区域，并确定所述热力区域中各像素点对应的热力权重；

计算所述热力区域中像素点的坐标与所述第一特征点热力图中像素点的坐标间的距离；

根据所述热力权重对所述距离进行加权求和，并将求和结果作为所述损失值。

4.根据权利要求1所述的热力图转换模型训练方法，所述根据目标特征点坐标确定所述目标特征点坐标对应的热力区域，并基于所述热力区域生成所述目标特征点对应的第一特征点热力图，包括：

以目标特征点坐标为圆心，以预设距离阈值为半径，确定所述目标特征点坐标对应的热力区域；

根据所述热力区域中各像素点对应的热力权重计算所述热力区域中各像素点的热力值；

根据所述热力区域中每个像素点的热力值，在所述热力区域中呈现所述各像素点的热力值对应的颜色，生成所述热力区域的第一特征点热力图。

5.根据权利要求1所述的热力图转换模型训练方法，所述根据目标特征点坐标确定所述目标特征点坐标对应的热力区域，并基于所述热力区域生成所述目标特征点对应的第一特征点热力图，包括：

以目标特征点坐标为中心，以预设距离阈值为边长，确定所述目标特征点坐标对应的方形热力区域；

根据所述方形热力区域中各像素点对应的热力权重计算所述方形热力区域中各像素点的热力值；

根据所述方形热力区域中每个像素点的热力值，在所述方形热力区域中呈现所述各像素点的热力值对应的颜色，生成所述方形热力区域的第一特征点热力图。

6.根据权利要求1所述的热力图转换模型训练方法，所述根据目标特征点坐标确定所述目标特征点坐标对应的热力区域，并基于所述热力区域生成所述目标特征点对应的第一特征点热力图，包括：

以目标特征点坐标为圆心，以预设距离阈值为半径，确定所述目标特征点坐标对应的热力区域；

为所述目标特征点坐标设置初始热力值，并将所述初始热力值作为均值；

将所述均值以及预设的方差输入热力值计算函数，计算获得所述热力区域中各像素点的热力值；

根据所述热力区域中每个像素点的热力值，在所述热力区域中呈现所述各像素点的热力值对应的颜色，生成所述热力区域的第一特征点热力图。