[发明专利]一种图像识别方法、模型训练方法及装置

说明书

本申请实施例提供了一种图像识别方法、模型训练方法及装置，涉及图像处理技术领域，包括：将获取的待识别人脸图像输入至目标人脸识别模型，得到待识别人脸图像的图像特征和待识别人脸图像中的人脸佩戴口罩的概率；计算待识别人脸图像的图像特征与模板人脸图像的图像特征的相似度，得到待识别人脸图像与模板人脸图像的相似度；如果待识别人脸图像中的人脸佩戴口罩的概率大于概率阈值，减小相似度阈值得到调整后的相似度阈值；在待识别人脸图像与模板人脸图像的相似度大于调整后的相似度阈值时，确定待识别人脸图像属于模板人脸图像所属的目标用户，待识别人脸图像中的人脸佩戴口罩时可以识别出待识别人脸图像所属的用户，提高人脸识别的准确性。

技术领域

本申请涉及图像处理技术领域，特别是涉及一种图像识别方法、模型训练方法及装置。

背景技术

为了保证目标场所的安全，在用户进入目标场所时需要进行人脸识别，并基于人脸识别的结果进行安全验证。例如，在用户需要进入学校时，采集用户的人脸图像(可以称为待识别人脸图像)，并基于多个模板人脸图像，对待识别人脸图像进行人脸识别，多个模板人脸图像为预先采集的校内人员的人脸图像。如果识别结果表示待识别人脸图像属于校内人员，则确定该用户通过安全验证，并允许该用户进入学校；如果识别结果表示待识别人脸图像属于陌生人员，则确定该用户未通过安全验证，并禁止该用户进入学校。

相关技术中，在对待识别人脸图像进行人脸识别时，计算待识别人脸图像与模板人脸图像的相似度，如果待识别人脸图像与一个模板人脸图像的相似度大于相似度阈值，确定待识别人脸图像属于模板人脸图像所属的用户。如果待识别人脸图像与各模板人脸图像的相似度均不大于相似度阈值，则确定待识别人脸图像属于陌生用户。

然而，用户可能会佩戴口罩，则待识别人脸图像中的人脸是佩戴口罩的，而模板人脸图像中人脸是未佩戴口罩的，基于未佩戴口罩的模板人脸图像进行识别，导致无法识别出待识别人脸图像所属的用户。

发明内容

本申请实施例的目的在于提供一种图像识别方法、模型训练方法及装置，以实现在待识别人脸图像中的人脸佩戴口罩时也可以识别出待识别人脸图像所属的用户，提高人脸识别的准确性。具体技术方案如下：

在本申请实施例的第一方面，提供了一种图像识别方法，所述方法包括：

获取待识别人脸图像；

将所述待识别人脸图像输入至预先训练的目标人脸识别模型，得到所述目标人脸识别模型输出的所述待识别人脸图像的图像特征，以及所述待识别人脸图像中的人脸佩戴口罩的概率；

计算所述待识别人脸图像的图像特征与预设的模板人脸图像的图像特征的相似度，得到所述待识别人脸图像与所述模板人脸图像的相似度；其中，所述模板人脸图像为预先采集的目标用户的人脸图像；

如果所述待识别人脸图像中的人脸佩戴口罩的概率大于预设的概率阈值，减小预设的相似度阈值，得到调整后的相似度阈值；

在所述待识别人脸图像与所述模板人脸图像的相似度大于所述调整后的相似度阈值时，确定所述待识别人脸图像属于所述模板人脸图像所属的目标用户。

可选的，所述获取待识别人脸图像，包括：

获取待处理图像；

基于预先训练的目标人脸检测模型对所述待处理图像进行人脸检测，得到所述待处理图像中包含人脸图像的图像区域，作为待识别人脸图像。

可选的，所述基于预先训练的目标人脸检测模型对所述待处理图像进行人脸检测，得到所述待处理图像中包含人脸图像的图像区域，作为待识别人脸图像，包括：

将所述待处理图像输入至预先训练的目标人脸检测模型，得到所述目标人脸检测模型输出的所述待处理图像中包含人脸图像的备选图像区域，以及每一备选图像区域中包含人脸图像的概率；