CurricularFace人脸识别模型不可以用本地的图片进行判断吗?
你好,是可以的,在代码示例中使用自己的图片,这里内部集成了SCRFD人脸检测 ,所以本地图片直接传原图即可,会比较两张图中的最大脸的相似度。
import cv2
import numpy as np
from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks
from modelscope.outputs import OutputKeys
img1 = 'your/img/1.jpg' # <---这里替换
img2 = 'your/img/2.jpg'
face_recognition = pipeline(Tasks.face_recognition, model='damo/cv_ir101_facerecognition_cfglint')
emb1 = face_recognition(img1)[OutputKeys.IMG_EMBEDDING]
emb2 = face_recognition(img2)[OutputKeys.IMG_EMBEDDING]
sim = np.dot(emb1[0], emb2[0])
print(f'Face cosine similarity={sim:.3f}, img1:{img1} img2:{img2}')