人脸识别后处理流程代码

人脸注册

tom = [...]  # (1, 128)
jack = [...]  # (1, 128)

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

人脸加载

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

人脸比对

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

人脸识别

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

应用

names, embeddings = load_embeddings('faces.npz')
while 1:
	faces = [...]
	for face in faces:
		embedding = [...]
		name = recognize(embedding, embeddings, names)
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值