稳定调用及效果更好的API,详见视觉开放智能平台:人脸比对1:1、口罩人脸比对1:1、人脸搜索1:N、公众人物识别、明星识别。
口罩人脸识别模型FRFM。
口罩人脸识别模型FRFM基于ArcFace框架,网络结构为NAS搜索出来的large model,训练数据使用了Glint360数据,同时加了数据增强的策略,包括random erasing,cutout策略和以一定概率增加口罩, 大大增强了在口罩数据集上的精度, 在MFR数据集中性能可以在96以上,同时在标准人脸识别任务上的精度也有提高。
本模型可以检测输入图片中带口罩和不戴口罩人脸的特征
from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks
from modelscope.outputs import OutputKeys
import numpy as np
face_mask_recognition_func = pipeline(Tasks.face_recognition, 'damo/cv_manual_face-recognition_frfm')
img1 = 'https://modelscope.oss-cn-beijing.aliyuncs.com/test/images/face_recognition_1.png'
img2 = 'https://modelscope.oss-cn-beijing.aliyuncs.com/test/images/face_recognition_2.png'
emb1 = face_mask_recognition_func(img1)[OutputKeys.IMG_EMBEDDING]
emb2 = face_mask_recognition_func(img2)[OutputKeys.IMG_EMBEDDING]
sim = np.dot(emb1[0], emb2[0])
print(f'Face cosine similarity={sim:.3f}, img1:{img1} img2:{img2}')
Method | IJBC(1e-5) | IJBC(1e-4) | MFR-ALL |
---|---|---|---|
FaceMask | - | - | 96+ |
本模型及代码来自达摩院和Insightface联合研发技术