[发明专利]基于人工智能的宠物分类方法、装置、设备及存储介质

说明书

本申请涉及人工智能中的生物识别领域，提供了一种宠物分类方法，包括：获取宠物图像和目标宠物分类模型；通过将宠物图像输入特征提取网络得到多个分辨率不同的目标特征图；将目标特征图输入面部框预测网络得到面部框的目标位置点组；将目标特征图输入第一宠物分类网络得到宠物的类型为各预设宠物类型的第一输出概率；从宠物图像中获取目标位置点组对应的宠物脸图像；将宠物脸图像输入第二宠物分类网络得到宠物的类型为各预设宠物类型的第二输出概率；根据第一输出概率和第二输出概率确定宠物的目标类型。该方法提高了宠物类型的识别准确性。本申请还涉及区块链技术，上述目标宠物分类模型可存储于区块链中。

技术领域

本申请涉及生物识别的技术领域，尤其涉及一种基于人工智能的宠物分类方法、装置、设备及存储介质。

背景技术

目前，用户可以给自家的宠物投保宠物险，当自己的宠物生病或出现意外时，可以向保险公司申请理赔，可以保障宠物的安全和减少自己的经济损失。投保宠物险时，需要对宠物进行身份认证和宠物类型识别，现有的DNA认证、芯片认证等手段操作复杂，成本较高。为解决上述问题，可以通过神经网络模型来识别宠物类型，然而现有的神经网络模型的准确性较低。因此，如何提高宠物类型的识别准确性是目前亟待解决的问题。

发明内容

本申请实施例提供一种基于人工智能的宠物分类方法、装置、设备及存储介质，旨在提高宠物类型的识别准确性。

第一方面，本申请实施例提供一种基于人工智能的宠物分类方法，包括：

获取宠物图像并调用目标宠物分类模型，所述目标宠物分类模型包括特征提取网络、面部框预测网络、第一宠物分类网络和第二宠物分类网络；

将所述宠物图像输入所述特征提取网络进行特征提取，得到多个分辨率不同的目标特征图；

将所述目标特征图输入所述面部框预测网络进行面部框预测，得到所述宠物图像的面部框的目标位置点组；

将所述目标特征图输入所述第一宠物分类网络进行分类，得到所述宠物图像中的宠物的类型为预设宠物类型的第一输出概率；

从所述宠物图像中获取所述目标位置点组对应的宠物脸图像，并将所述宠物脸图像输入所述第二宠物分类网络进行分类，得到所述宠物的类型为所述预设宠物类型的第二输出概率；

根据所述第一输出概率和所述第二输出概率，从各所述预设宠物类型中确定所述宠物图像中的宠物的目标类型。

第二方面，本申请实施例还提供一种宠物分类装置，所述宠物分类装置包括：

获取模块，用于获取宠物图像并调用目标宠物分类模型，所述目标宠物分类模型包括特征提取网络、面部框预测网络、第一宠物分类网络和第二宠物分类网络；

特征提取模块，用于将所述宠物图像输入所述特征提取网络进行特征提取，得到多个分辨率不同的目标特征图；

面部框预测模块，用于将所述目标特征图输入所述面部框预测网络进行面部框预测，得到所述宠物图像的面部框的目标位置点组；

第一宠物分类模块，用于将所述目标特征图输入所述第一宠物分类网络进行分类，得到所述宠物图像中的宠物的类型为预设宠物类型的第一输出概率；

确定模块，用于从所述宠物图像中获取所述目标位置点组对应的宠物脸图像；

第二宠物分类模块，用于将所述宠物脸图像输入所述第二宠物分类网络进行分类，得到所述宠物图像中的宠物的类型为所述预设宠物类型的第二输出概率；

所述确定模块，还用于根据所述第一输出概率和所述第二输出概率，从各所述预设宠物类型中确定所述宠物图像中的宠物的目标类型。