[发明专利]目标图像的关键点检测方法、装置、电子设备及存储介质

权利要求书

1.一种目标图像的关键点检测方法，其特征在于，包括：

获取训练数据；

将所述训练数据输入第一网络训练得到第一网络模型；

将所述训练数据输入第二网络，根据所述第一网络模型对所述第二网络进行蒸馏学习，在所述蒸馏学习的同时对所述第二网络进行剪枝处理，得到关键点检测模型，其中，所述蒸馏学习中的知识蒸馏损失函数包括标志函数，所述标志函数用于标志所述训练数据中的正确样本和错误样本；

将待检测的目标图像输入所述关键点检测模型，得到所述目标图像中的关键点。

2.根据权利要求1所述的一种目标图像的关键点检测方法，其特征在于，所述将所述训练数据输入第一网络训练得到第一网络模型，包括：

将所述训练数据输入所述第一网络进行可分离卷积处理，得到不同尺度的第一特征图集合；所述训练数据具有标签，所述标签为所述训练数据的关键点真实坐标值；

通过所述第一网络的全连接层对所述第一特征图集合进行第一关键点坐标值预测，得到第一预测结果；

采用第一损失函数确定所述第一预测结果和所述标签之间的第一损失值；

根据所述第一损失值修改第一网络模型参数，并根据修改后的所述第一网络模型参数生成所述第一网络模型。

3.根据权利要求2所述的一种目标图像的关键点检测方法，其特征在于，所述通过所述第一网络的全连接层对所述第一特征图集合进行第一关键点坐标值预测，得到第一预测结果这一步骤之前，还包括：

对所述第一特征图集合中各个第一特征图进行全局平均池化处理，得到各个所述第一特征图的第一通道权重。

4.根据权利要求2所述的一种目标图像的关键点检测方法，其特征在于，所述将所述训练数据输入第二网络，根据所述第一网络模型对所述第二网络进行蒸馏学习，在所述蒸馏学习的同时对所述第二网络进行剪枝处理，得到关键点检测模型，包括：

将所述训练数据输入所述第二网络进行可分离卷积处理，得到不同尺度的第二特征图集合；

通过所述第二网络的全连接层对所述第二特征图集合进行第二关键点坐标值预测，得到第二预测结果；

采用第二损失函数确定所述第二预测结果和所述标签之间的第二损失值；

根据所述第二损失值和所述知识蒸馏损失函数修改第二网络模型参数，并根据修改后的所述第二网络模型参数生成第二网络模型；

对所述第二网络模型进行剪枝处理，得到所述关键点检测模型。

5.根据权利要求4所述的一种目标图像的关键点检测方法，其特征在于，所述根据所述第二损失值和所述知识蒸馏损失函数修改第二网络模型参数，并根据修改后的所述第二网络模型参数生成第二网络模型，包括：

根据所述第一预测结果确定第一预测分布，并根据所述第二预测结果确定第二预测分布；

采用所述知识蒸馏损失函数确定所述第一预测分布与所述第二预测分布之间的第三损失值；

根据所述第二损失值和所述第三损失值修改第二网络模型参数，并根据修改后的所述第二网络模型参数生成第二网络模型。

6.根据权利要求5所述的一种目标图像的关键点检测方法，其特征在于，所述采用所述知识蒸馏损失函数确定所述第一预测分布与所述第二预测分布之间的第三损失值，包括：

确定所述训练数据中的正确样本；

将所述第一预测分布中正确样本对应的标志函数置1，并将所述第一预测分布中错误样本对应的标志函数置0，得到赋值后的标志函数；

根据所述赋值后的标志函数确定知识蒸馏损失函数，并采用所述知识蒸馏损失函数计算所述第一预测分布与所述第二预测分布之间的差异值，得到所述第三损失值。

7.根据权利要求4所述的一种目标图像的关键点检测方法，其特征在于，所述对所述第二网络模型进行剪枝处理，得到所述关键点检测模型这一步骤之前，还包括：

对所述第二特征图集合中各个第二特征图进行全局平均池化处理，得到各个所述第二特征图的第二通道权重。