首页 > 其他分享 >21 人脸识别

21 人脸识别

时间:2023-10-08 17:13:19浏览次数:33  
标签:编码 人脸识别 dist 21 database 人脸 图像 函数

import tensorflow as tf
from tensorflow.keras import Sequential
from tensorflow.keras.models import Model
from tensorflow.keras.layers import Conv2D, ZeroPadding2D, Activation, Input, concatenate
from tensorflow.keras.layers import BatchNormalization, MaxPooling2D, AveragePooling2D, Concatenate
from tensorflow.keras.layers import Lambda, Flatten, Dense
from tensorflow.keras.initializers import GlorotUniform
#from tensorflow.keras.engine.topology import Layer
from tensorflow.keras import backend as K
K.set_image_data_format('channels_last')
import imageio
import os
import numpy as np
from numpy import genfromtxt
#import pandas as pd

from fr_utils import *
from inception_blocks_v2 import *

%matplotlib inline
#%load_ext autoreload
#%autoreload 2

#np.set_printoptions(threshold=np.nan)

这段代码是一个TensorFlow的深度学习模型的导入和配置部分。

  1. 导入了TensorFlow的相关库。
  2. 导入了自定义的一些模块,例如fr_utilsinception_blocks_v2,这些模块可能包含了一些自定义的深度学习层或函数。
  3. 设置了TensorFlow的图像数据格式为'channels_last'。
  4. 导入了一些其他的Python库,如NumPy和imageio。

这段代码还包含了一些Jupyter Notebook的魔法命令,例如%matplotlib inline,它用于在Jupyter Notebook中显示绘图。

此外,代码中还有一些注释,但是由于没有提供fr_utils.pyinception_blocks_v2.py的代码,因此无法确定这些模块的具体内容和功能。这些模块可能包含了与人脸识别或相关任务有关的深度学习模型和功能。

FRmodel = faceRecoModel(input_shape=(96, 96, 3))

这行代码创建了一个名为FRmodel的人脸识别模型。这个模型的输入形状是(96, 96, 3),表示输入图像的尺寸为96x96像素,通道数为3(对应RGB彩色图像)。

faceRecoModel可能是一个自定义的人脸识别模型,根据输入的图像进行人脸识别任务。该模型可能由多个卷积层、池化层、全连接层等组成,以便对输入图像进行特征提取和识别。

print("Total Params:", FRmodel.count_params())

这行代码打印了FRmodel模型的总参数数量,即模型中可学习的参数的数量。参数数量是衡量深度学习模型复杂度的一种指标,通常与模型的规模和容量相关。

通过打印参数数量,可以了解模型的大小以及需要训练的参数数量。这对于模型的内存占用和计算资源要求等方面的估计非常有用。

# 实现三元组损失函数

def triplet_loss(y_true, y_pred, alpha = 0.2):
    """
    参数:
    y_true -- 这个参数我们暂时不用
    y_pred -- 这是一个python列表,里面包含了3个对象,分别是A,P,N的编码。
    
    返回值:
    loss -- 损失值
    """
    
    anchor, positive, negative = y_pred[0], y_pred[1], y_pred[2]

    # 计算A与P的编码差异,就是公式中标(1)的那块
    pos_dist = tf.reduce_sum(tf.square(tf.subtract(anchor, positive)))
    # 计算A与N的编码差异,就是公式中标(2)的那块
    neg_dist = tf.reduce_sum(tf.square(tf.subtract(anchor, negative)))
    # 根据两组编码差异来计算损失
    basic_loss = tf.add(tf.subtract(pos_dist, neg_dist), alpha)
    loss = tf.maximum(tf.reduce_mean(basic_loss), 0.0)
    
    return loss

这段代码实现了一个三元组损失函数 triplet_loss,用于训练人脸识别模型。这种损失函数通常用于将同一个人的人脸图像编码后的特征向量更近地靠拢,而将不同人的人脸图像编码后的特征向量更远地分开。

  • y_true:这个参数目前没有被使用,通常在Keras的损失函数中需要,但在这个实现中没有使用。

  • y_pred:这个参数是一个Python列表,其中包含3个对象,分别是anchorpositivenegative的编码。这三个编码分别表示锚定图像(anchor image)、正样本图像(positive image)和负样本图像(negative image)的特征编码。

  • alpha:这是一个超参数,用于控制编码之间的间隔。它表示在计算编码之间的差异时,允许正样本和负样本之间的距离小于alpha

接下来,损失函数计算了以下几个步骤:

  1. 计算锚定图像(A)和正样本图像(P)的编码差异(pos_dist):它是两个特征向量之间的平方欧氏距离。

  2. 计算锚定图像(A)和负样本图像(N)的编码差异(neg_dist):同样是两个特征向量之间的平方欧氏距离。

  3. 计算基本损失(basic_loss):这是pos_distneg_dist之间的差异,再加上alpha。这个差异表示了正样本和负样本之间的距离差异,并且保证了正样本之间的距离要小于alpha

  4. 最后,通过使用 tf.maximum 将基本损失与0比较,确保损失值不小于0。然后,计算平均损失值。

这个三元组损失函数的目标是最小化同一个人的人脸编码之间的距离(锚定图像与正样本图像),并最大化不同人的人脸编码之间的距离(锚定图像与负样本图像)。通过调整超参数alpha,可以控制编码之间的间隔。

tf.random.set_seed(1)
y_true = (None, None, None)
y_pred = (tf.random.normal([3, 128], mean=6, stddev=0.1),
          tf.random.normal([3, 128], mean=1, stddev=1),
          tf.random.normal([3, 128], mean=3, stddev=4))
loss = triplet_loss(y_true, y_pred)
    
tf.print("loss = ", loss)

这段代码使用了之前定义的triplet_loss函数来计算三元组损失。以下是代码的执行过程和结果:

  1. tf.random.set_seed(1):这行代码设置了TensorFlow的随机种子,以确保随机生成的数值在每次执行时保持一致。

  2. y_true = (None, None, None):在这里,y_true被设置为None,但在triplet_loss函数中没有使用,因为该函数的实现中并没有使用y_true参数。

  3. y_pred:这个变量包含了三个特征编码向量,分别代表锚定图像、正样本图像和负样本图像的编码。这些编码是通过使用tf.random.normal来生成的,具有不同的均值和标准差。

  4. loss = triplet_loss(y_true, y_pred):这一行代码调用了triplet_loss函数来计算三元组损失,传递了y_truey_pred作为参数。

  5. tf.print("loss = ", loss):最后,通过tf.print打印了计算得到的损失值。

这段代码的目的是演示如何使用triplet_loss函数来计算三元组损失,但由于y_true没有在函数中使用,因此实际上并没有使用真正的标签。在实际应用中,y_true通常包含真实的标签信息,以便进行监督学习任务。

您可以根据自己的需求替换y_truey_pred,以计算实际人脸识别任务中的三元组损失。

FRmodel.compile(optimizer = 'adam', loss = triplet_loss, metrics = ['accuracy'])
load_weights_from_FaceNet(FRmodel)

这段代码涉及到人脸识别模型的编译和加载权重的过程

  1. FRmodel.compile(optimizer='adam', loss=triplet_loss, metrics=['accuracy']):这行代码编译了FRmodel模型。具体地:

    • optimizer='adam':选择了Adam优化器作为训练模型的优化器。
    • loss=triplet_loss:指定了损失函数为之前定义的triplet_loss函数。这是人脸识别任务的自定义损失函数,用于训练模型以使编码向量满足三元组约束。
    • metrics=['accuracy']:设置了评估指标,这里选择了准确率作为评估指标。
  2. load_weights_from_FaceNet(FRmodel):这行代码调用了一个自定义的函数load_weights_from_FaceNet,该函数的作用是加载预训练的权重到FRmodel模型中。这种预训练权重通常是在大规模人脸识别数据集上训练的,可以提供更好的初始参数以加速模型的收敛。

总之,这段代码准备了FRmodel模型以进行人脸识别任务的训练。它指定了优化器、损失函数和评估指标,并加载了预训练的权重,以便模型具有良好的初始参数。

database = {}
database["danielle"] = img_to_encoding("images/danielle.png", FRmodel)
database["younes"] = img_to_encoding("images/younes.jpg", FRmodel)
database["tian"] = img_to_encoding("images/tian.jpg", FRmodel)
database["andrew"] = img_to_encoding("images/andrew.jpg", FRmodel)
database["kian"] = img_to_encoding("images/kian.jpg", FRmodel)
database["dan"] = img_to_encoding("images/dan.jpg", FRmodel)
database["sebastiano"] = img_to_encoding("images/sebastiano.jpg", FRmodel)
database["bertrand"] = img_to_encoding("images/bertrand.jpg", FRmodel)
database["kevin"] = img_to_encoding("images/kevin.jpg", FRmodel)
database["felix"] = img_to_encoding("images/felix.jpg", FRmodel)
database["benoit"] = img_to_encoding("images/benoit.jpg", FRmodel)
database["arnaud"] = img_to_encoding("images/arnaud.jpg", FRmodel)

这段代码创建了一个名为database的字典,用于存储人脸图像的编码向量。每个键(如"danielle"、"younes"等)代表一个人的名字,对应的值是该人脸图像的编码向量。编码向量是通过调用img_to_encoding函数生成的,该函数将人脸图像转换为特定模型(FRmodel)生成的编码向量。

这个数据库似乎用于存储已知人脸的编码,以便在后续的人脸识别任务中使用。具体来说,每个人的编码向量将用于与输入的人脸图像进行比对,以确定输入图像中是否存在已知的人脸。

# 验证人脸

def verify(image_path, identity, database, model):
    """
    
    参数:
    image_path -- 需要被验证的人脸图片地址,也就是摄像头拍摄到的人脸图片
    identity -- 人名,也就是扫描ID后得到的人名
    database -- 人脸数据库
    model -- 前面加载的已经训练好了的FaceNet模型
    
    返回值:
    dist -- 返回image_path的人脸编码与identity指定的人名的人脸编码的差异
    door_open -- 如果返回True, 就表示image_path和identity是同一个人,那么验证通过,门自动为他开启
    """
    
    encoding = img_to_encoding(image_path, model)
    
    # 对比identity关联的图片编码与image_path关联的图片编码的差异
    dist = np.linalg.norm(encoding-database[identity])
    
    # 差异小于0.7,就表明是同一个人。
    # 当然,因为我们的程序的目的是教学,所以精度不高,实际应用中会取比0.7更小的阈值
    if dist < 0.7:
        print("It's " + str(identity) + ", welcome home!")
        door_open = True
    else:
        print("It's not " + str(identity) + ", please go away")
        door_open = False
        
    return dist, door_open

这段代码定义了一个名为verify的函数,用于验证输入的人脸图像是否属于指定的身份(identity)。

  • image_path:需要被验证的人脸图像的文件路径,通常是通过摄像头拍摄到的人脸图像。

  • identity:人名,即扫描ID后得到的人名,表示要验证的身份。

  • database:人脸数据库,存储了已知人脸的编码向量,用于与输入图像进行比对。

  • model:已经加载并训练好的FaceNet模型,用于将输入图像转换为编码向量。

函数的主要工作包括:

  1. 使用img_to_encoding函数将输入的人脸图像(image_path)转换为编码向量(encoding)。

  2. 计算输入图像的编码向量与指定身份的编码向量之间的欧氏距离(dist)。

  3. 判断欧氏距离是否小于0.7(这个阈值可以根据需要调整)。如果距离小于0.7,函数会打印验证通过的消息,并返回door_open为True,表示门应该自动打开。否则,函数会打印验证失败的消息,并返回door_open为False,表示门不会打开。

总之,这个函数用于验证输入的人脸图像是否与指定身份的人脸匹配。如果匹配成功,门会自动打开;如果匹配失败,门不会打开。

verify("images/camera_0.jpg", "younes", database, FRmodel)

这行代码调用了之前定义的verify函数,用于验证位于文件路径"images/camera_0.jpg"的人脸图像是否属于身份"younes"。下面是代码的执行过程和可能的结果:

  1. 首先,函数使用img_to_encoding函数将输入的人脸图像"images/camera_0.jpg"转换为编码向量。

  2. 然后,函数计算输入图像的编码向量与数据库中身份"younes"对应的编码向量之间的欧氏距离。

  3. 如果欧氏距离小于0.7,函数会打印消息"It's younes, welcome home!",并将door_open设置为True,表示门自动打开。

  4. 如果欧氏距离大于等于0.7,函数会打印消息"It's not younes, please go away",并将door_open设置为False,表示门不会打开。

最终,函数会返回欧氏距离(dist)和门的开启状态(door_open)。

请注意,欧氏距离的阈值0.7是一个示例值,用于演示目的。在实际应用中,您可能需要根据实际情况和精度要求来调整这个阈值。

verify("images/camera_2.jpg", "kian", database, FRmodel)

这行代码调用了verify函数,用于验证位于文件路径"images/camera_2.jpg"的人脸图像是否属于身份"kian"。根据verify函数的逻辑,以下是可能的结果:

  • 如果人脸图像"images/camera_2.jpg"与身份"kian"的编码非常相似(欧氏距离小于0.7),则函数会打印消息"It's kian, welcome home!",并将door_open设置为True,表示门应该自动打开。

  • 如果人脸图像"images/camera_2.jpg"与身份"kian"的编码差异较大(欧氏距离大于等于0.7),则函数会打印消息"It's not kian, please go away",并将door_open设置为False,表示门不会打开。

根据具体的欧氏距离,上述的两种情况中的一种将会发生。

请注意,这个代码示例用于演示人脸验证的基本原理,实际应用中可能需要更复杂的流程和更精确的距离阈值来确保安全性和准确性。

# 人脸识别

def who_is_it(image_path, database, model):
    """    
    参数:
    image_path -- 需要被识别的人脸图像的地址
    database -- 有权开门的人脸数据库
    model -- 前面加载的已经训练好了的FaceNet模型
    
    Returns:
    min_dist -- 最小差异。我们会一个个与数据库中的人脸编码进行对比,找到与待识别的图像差异最小的那个图像
    identity -- 差异最小的人脸图像关联的那个人的名字
    """
    
    encoding = img_to_encoding(image_path, model)
    
    min_dist = 100
    
    # 与数据库中的人脸一个个的进行对比
    for (name, db_enc) in database.items():
        
        # 进行对比
        dist = np.linalg.norm(encoding-db_enc)

        # 保存差异最小的那个
        if dist < min_dist:
            min_dist = dist
            identity = name
    
    if min_dist > 0.7:
        print("Not in the database.")
    else:
        print ("it's " + str(identity) + ", the distance is " + str(min_dist))
        
    return min_dist, identity

这段代码定义了一个名为who_is_it的函数,用于识别输入的人脸图像属于数据库中的哪个人。

  • image_path:需要被识别的人脸图像的文件路径。

  • database:包含有权开门的人脸数据库,其中每个键是一个人的名字,对应的值是该人的人脸编码向量。

  • model:已经加载并训练好的FaceNet模型,用于将输入图像转换为编码向量。

函数的主要工作包括:

  1. 使用img_to_encoding函数将输入的人脸图像(image_path)转换为编码向量(encoding)。

  2. 初始化min_dist为一个较大的值(例如100),用于保存差异最小的人脸编码的欧氏距离。

  3. 遍历数据库中的每个人脸编码,计算输入图像的编码向量与数据库中每个编码向量的欧氏距离。

  4. 如果找到了欧氏距离小于min_dist的编码,就将min_dist更新为找到的最小距离,并将identity设置为对应的人名。

  5. 最后,根据最小距离(min_dist)判断输入图像是否属于数据库中的某个人。如果最小距离大于0.7,函数会打印"Not in the database."的消息;否则,会打印"it's [identity]"以及距离信息。

函数返回最小距离(min_dist)和被识别的人的名字(identity)。

请注意,欧氏距离的阈值0.7是一个示例值,用于演示目的。在实际应用中,您可能需要根据具体情况来调整此阈值以获得更好的识别精度。

who_is_it("images/camera_0.jpg", database, FRmodel)

这行代码调用了who_is_it函数,用于识别位于文件路径"images/camera_0.jpg"的人脸图像属于数据库中的哪个人。根据who_is_it函数的逻辑,以下是可能的结果:

  • 如果人脸图像"images/camera_0.jpg"与数据库中的某个人脸编码非常相似(欧氏距离小于0.7),函数会打印消息"it's [identity]",其中[identity]是与输入图像最相似的人的名字,以及距离信息。

  • 如果人脸图像"images/camera_0.jpg"与数据库中的所有人脸编码都差异较大(欧氏距离大于等于0.7),函数会打印消息"Not in the database.",表示没有找到匹配的人脸。

请注意,欧氏距离的阈值0.7是一个示例值,用于演示目的。在实际应用中,您可能需要根据具体情况来调整此阈值以获得更好的识别精度。

标签:编码,人脸识别,dist,21,database,人脸,图像,函数
From: https://www.cnblogs.com/wzbzk/p/17749628.html

相关文章

  • 【LCD驱动】VK1C21系列是防静电/抗干扰LCD液晶显示段码驱动芯片,可驱动32*4/18*4/14*4
    产品型号:VK1C21A/B产品品牌:永嘉微电/VINKA封装形式:SSOP48/LQFP48可定制裸片:DICE(COB邦定片);COG(邦定玻璃用)产品年份:新年份原厂,工程服务,技术支持! 概述:VK1C21A/B是一个点阵式存储映射的LCD驱动器,可支持最大128点(32SEGx4COM)的LCD屏,也支持2COM和3COM的LCD屏。单片机可通过......
  • JDK 21新特性---有序集合
    概述在JDK21中,正式加入了序列集合的功能,新增了SequencedCollection,SequencedSet,SequencedMap接口,使得Java中的有序集合类可以按照统一的方法来进行集合的操作。架构图:三大接口SequencedCollectioninterfaceSequencedCollection<E>extendsCollection<E>{......
  • 2023-2024-1 20231421 《计算机基础与程序设计》第二周学习总结
    作业信息作业要求:https://www.cnblogs.com/rocedu/p/9577842.html#WEEK02作业目标:自学《计算机科学概论》和《c语言程序设计》第一章教材学习内容总结一、《计算机科学概论》1.计算机系统是由信息,硬件,程序设计,操作系统,应用和通信由内而外组成的2.了解了计算机的历史二、《c......
  • 215. 数组中的第K个最大元素
    给定整数数组nums和整数k,请返回数组中第k个最大的元素。请注意,你需要找的是数组排序后的第k个最大的元素,而不是第k个不同的元素。你必须设计并实现时间复杂度为O(n)的算法解决此问题。示例1:输入:[3,2,1,5,6,4],k=2输出:5代码classSolution{publ......
  • 【LCD驱动】VK1C21系列是防静电/抗干扰LCD液晶显示驱动芯片,可驱动32*4/18*4/14*4点
    产品型号:VK1C21A/B产品品牌:永嘉微电/VINKA封装形式:SSOP48/LQFP48可定制裸片:DICE(COB邦定片);COG(邦定玻璃用)产品年份:新年份原厂,工程服务,技术支持! 概述:VK1C21A/B是一个点阵式存储映射的LCD驱动器,可支持最大128点(32SEGx4COM)的LCD屏,也支持2COM和3COM的LCD屏。单片机可通过......
  • slices in Go 1.21
    Go1.21中新增的slices包中提供了很多与切片相关的函数,适用于任意类型的切片。本文内容来自官方文档BinarySearch函数签名如下:funcBinarySearch[S~[]E,Ecmp.Ordered](xS,targetE)(int,bool)BinarySearch在已排序的切片中搜索target并返回找到target的位置,或......
  • JKD21的虚拟线程你会用了吗?
    大家好,我是老七,关注我,将持续更新更多精彩内容!Java21正式版已经问世,作为继Java17之后的又一款长期支持版(LTS),为我们带来了一场技术盛宴。其中,最引人注目的新特性便是虚拟线程。想必大家对这个新的线程模型也是充满了好奇心。在闲暇之余,我已亲自安装了JDK21来一探究竟,并整理分享一......
  • 21207328-吴义隆-blog1
    一、前言:三次pta题目集知识点丰富,包括了一开始如何创建Java基本形式,接着相关的命名及其使用,以及后面的知识结合。并且三次题目集的题量还算中规中矩,一开始,较为基础,题目多一点,方便学生进行学习和理解;接着进阶一点,难度提升的同时,适当减少了习题量;到最后,基本上题目以第二题的课......
  • BLOG-1-7213
    (1)前言:主要用于完成学校的互评作业大家懂的都懂,前三次的题目集,大概囊括了最基础的输入输出、类的创建;字符串的各种操作(定位某个特点字符在字符串中的下标、提取某段需要的子字符串、对于某个字符或某个子字符串的格式判断等等)、类的交互、函数的创建与使用以及正则表达式的运用等等......
  • 【倍增】ABC212F Greedy Takahashi 题解
    ABC212F暴力就是直接跳,显然不可过。考虑对暴力进行优化,发现整个图是不会改变的,容易想到使用倍增。显然是对边进行倍增的,令\(f_{i,j}\)表示从第\(i\)条边开始,跳了\(2^j\)条边后,到的是哪一条边,如果不存在,则设为\(-1\)。然后就是很显然的倍增了,最后讨论一下即可。时间复......