人脸识别后处理流程代码人脸注册人脸加载人脸比对人脸识别应用

815 阅读 0 评论 538 点赞

我是靠谱客的博主勤劳楼房，这篇文章主要介绍人脸识别后处理流程代码人脸注册人脸加载人脸比对人脸识别应用，现在分享给大家，希望可以做个参考。

文章目录

人脸注册
人脸加载
人脸比对
人脸识别
应用

人脸注册

复制代码

1
2
3
4
5
tom = [...]  # (1, 128)
jack = [...]  # (1, 128)

np.savez('faces.npz', tom=tom, jack=jack)

人脸加载

复制代码

1
2
3
4
5
6
7
8
9
def load_embeddings(path):
    arrs = np.load(path)
    names, embeddings = [], []
    for name, embedding in arrs.items():
        names.append(name)
        embeddings.append(embedding)
    embeddings = np.vstack(embeddings)
    return np.array(names), embeddings

人脸比对

复制代码

1
2
3
4
def compare(embedding, embeddings):  # face: (1, 128), embeddings: (None, 128)
	# Euclidean distance
    return np.sqrt(np.sum(((embeddings - embedding) ** 2), axis=1))

人脸识别

复制代码

1
2
3
4
5
6
7
8
9
def recognize(embedding, embeddings, names):
    result = compare(embedding, embeddings)
    index = np.argsort(result)[::-1]
    score = result[index][0]
    if  score < 1.1:
        return names[index][0], score
    else:
        return 'unknown', score

应用

复制代码

1
2
3
4
5
6
7
names, embeddings = load_embeddings('faces.npz')
while 1:
	faces = [...]
	for face in faces:
		embedding = [...]
		name = recognize(embedding, embeddings, names)