口罩人脸识别模型FRFM-large
口罩人脸识别模型FRFM-large
  • 模型资讯
  • 模型资料

FRFM模型介绍

稳定调用及效果更好的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}')

使用方式

  • 推理:输入经过对齐的人脸图片(112x112),返回人脸特征向量(1024维),为便于体验,集成了人脸检测和关键点模型RetinaFace,输入两张图片,各自进行人脸检测选择最大脸并对齐后提取特征,然后返回相似度比分

目标场景

  • 人脸识别应用广泛,如考勤,通行,人身核验,智慧安防等场景

模型局限性及可能偏差

  • 由于模型较大,目前仅支持GPU推理。
  • 当前版本在python 3.7环境测试通过,其他环境下可用性待测试

模型性能指标

Method IJBC(1e-5) IJBC(1e-4) MFR-ALL
FaceMask - - 96+

来源说明

本模型及代码来自达摩院和Insightface联合研发技术