[发明专利]一种动物心情识别模型的训练方法和装置

权利要求书

1.一种动物心情识别模型的训练方法，其特征在于，包括：

获取训练集中的动物面部图像样本；

提取所述动物面部图像样本的关键点坐标；

将所述关键点坐标输入动物心情识别模型，得到所述动物心情识别模型输出的动物心情指数估计值；

将所述动物心情指数估计值与所述动物面部图像样本对应的动物心情指数标注值进行比较，根据所述比较的结果调整所述动物心情识别模型的参数。

2.根据权利要求1所述的方法，其特征在于，所述提取所述动物面部图像样本的关键点坐标，包括：

提取所述动物面部图像样本的多个关键点坐标；所述多个关键点坐标对应所述动物面部图像样本中的面部轮廓、眼角、眼眶最高点、眼眶最低点、眉毛轮廓、鼻子轮廓及嘴巴轮廓中的至少一项。

3.根据权利要求1或2所述的方法，其特征在于，所述根据所述比较的结果调整所述动物心情识别模型的参数，包括：

当所述比较的结果不满足预设要求时，根据所述比较的结果，采用反向传播和/或梯度下降方式，调整所述动物心情识别模型的参数；针对所述训练集中的下一个所述动物面部图像样本，返回执行所述获取训练集中的动物面部图像样本的步骤；

当所述比较的结果满足预设要求时，结束对所述动物心情识别模型的训练过程。

4.根据权利要求3所述的方法，其特征在于，所述结束对所述动物心情识别模型的训练过程之后，还包括对所述动物心情识别模型的验证过程；

所述验证过程包括：

采用验证集中的动物面部图像样本及其对应的动物心情指数标注值，计算所述动物心情识别模型的准确率和召回率；

根据所述准确率和召回率计算所述动物心情识别模型的ROC曲线；

在所述ROC曲线满足预设条件的情况下，结束对所述动物心情识别模型的验证过程。

5.一种动物心情识别方法，其特征在于，包括：

获取动物面部图像；

提取所述动物面部图像的关键点坐标；

将所述关键点坐标输入预先建立的动物心情识别模型；

获取所述动物心情识别模型输出的动物心情指数估计值；所述动物心情指数估计值由所述动物心情识别模型根据所述关键点坐标预测得出。

6.根据权利要求5所述的方法，其特征在于，所述提取所述动物面部图像的关键点坐标，包括：

提取所述动物面部图像的多个关键点坐标；所述多个关键点坐标对应所述动物面部图像中的面部轮廓、眼角、眼眶最高点、眼眶最低点、眉毛轮廓、鼻子轮廓及嘴巴轮廓中的至少一项。

7.根据权利要求5或6所述的方法，其特征在于，所述动物心情识别模型设置在服务器中；

所述将所述关键点坐标输入预先训练的动物心情识别模型，包括：向服务器发送心情识别请求消息，所述心情识别请求消息中携带所述关键点坐标；

所述获取所述动物心情识别模型输出的动物心情指数估计值，包括：接收所述服务器反馈的心情识别应答消息，所述心情识别应答消息中携带所述动物心情指数估计值；从所述心情识别应答消息中提取所述动物心情指数估计值。

8.一种动物心情识别模型的训练装置，其特征在于，包括：

第一获取模块，用于获取训练集中的动物面部图像样本；

第一提取模块，用于提取所述动物面部图像样本的关键点坐标；

第一输入模块，用于将所述关键点坐标输入动物心情识别模型，得到所述动物心情识别模型输出的动物心情指数估计值；

调整模块，用于将所述动物心情指数估计值与所述动物面部图像样本对应的动物心情指数标注值进行比较，根据所述比较的结果调整所述动物心情识别模型的参数。

9.根据权利要求8所述的装置，其特征在于，所述第一提取模块，用于提取所述动物面部图像样本的多个关键点坐标；所述多个关键点坐标对应所述动物面部图像样本中的面部轮廓、眼角、眼眶最高点、眼眶最低点、眉毛轮廓、鼻子轮廓及嘴巴轮廓中的至少一项。