"""Performs face alignment and calculates L2 distance between the embeddings of images."""

# MIT License

#

# Copyright (c) 2016 David Sandberg

#

# Permission is hereby granted, free of charge, to any person obtaining a copy

# of this software and associated documentation files (the "Software"), to deal

# in the Software without restriction, including without limitation the rights

# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell

# copies of the Software, and to permit persons to whom the Software is

# furnished to do so, subject to the following conditions:

#

# The above copyright notice and this permission notice shall be included in all

# copies or substantial portions of the Software.

#

# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR

# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,

# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE

# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER

# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,

# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE

# SOFTWARE.

from __future__ import absolute_import

from __future__ import division

from __future__ import print_function

from scipy import misc

import tensorflow as tf

import numpy as np

import sys

import os

import argparse

import facenet

import align.detect_face

def main():

    model = "../models/20170216-091149"

    image_files = ['compare_images/index10.png', 'compare_images/index73.png']

    image_size = 160

    margin = 44

    gpu_memory_fraction = 0.5

    images = load_and_align_data(image_files, image_size, margin, gpu_memory_fraction)

    with tf.Graph().as_default():

        with tf.Session() as sess:

            # Load the model

            facenet.load_model(model)

            # Get input and output tensors

            images_placeholder = tf.get_default_graph().get_tensor_by_name("input:0")

            embeddings = tf.get_default_graph().get_tensor_by_name("embeddings:0")

            phase_train_placeholder = tf.get_default_graph().get_tensor_by_name("phase_train:0")

            # Run forward pass to calculate embeddings

            feed_dict = { images_placeholder: images, phase_train_placeholder:False }

            emb = sess.run(embeddings, feed_dict=feed_dict)

            nrof_images = len(image_files)

            print('Images:')

            for i in range(nrof_images):

                print('%1d: %s' % (i, image_files[i]))

            print('')

            # Print distance matrix

            print('Distance matrix')

            print('    ', end='')

            for i in range(nrof_images):

                print('    %1d     ' % i, end='')

            print('')

            for i in range(nrof_images):

                print('%1d  ' % i, end='')

                for j in range(nrof_images):

                    dist = np.sqrt(np.sum(np.square(np.subtract(emb[i,:], emb[j,:]))))

                    print('  %1.4f  ' % dist, end='')

                print('')

def load_and_align_data(image_paths, image_size, margin, gpu_memory_fraction):

    minsize = 20 # minimum size of face

    threshold = [ 0.6, 0.7, 0.7 ]  # three steps's threshold

    factor = 0.709 # scale factor

    print('Creating networks and loading parameters')

    with tf.Graph().as_default():

        gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=gpu_memory_fraction)

        sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options, log_device_placement=False))

        with sess.as_default():

            pnet, rnet, onet = align.detect_face.create_mtcnn(sess, None)

    nrof_samples = len(image_paths)

    img_list = [None] * nrof_samples

    for i in range(nrof_samples):

        img = misc.imread(os.path.expanduser(image_paths[i]))

        img_size = np.asarray(img.shape)[0:2]

        bounding_boxes, _ = align.detect_face.detect_face(img, minsize, pnet, rnet, onet, threshold, factor)

        det = np.squeeze(bounding_boxes[0,0:4])

        bb = np.zeros(4, dtype=np.int32)

        bb[0] = np.maximum(det[0]-margin/2, 0)

        bb[1] = np.maximum(det[1]-margin/2, 0)

        bb[2] = np.minimum(det[2]+margin/2, img_size[1])

        bb[3] = np.minimum(det[3]+margin/2, img_size[0])

        cropped = img[bb[1]:bb[3],bb[0]:bb[2],:]

        aligned = misc.imresize(cropped, (image_size, image_size), interp='bilinear')

        prewhitened = facenet.prewhiten(aligned)

        img_list[i] = prewhitened

    images = np.stack(img_list)

    return images

# def parse_arguments(argv):

#     parser = argparse.ArgumentParser()

#

#     parser.add_argument('model', type=str, default="./models/20170216-091149",

#         help='Could be either a directory containing the meta_file and ckpt_file or a model protobuf (.pb) file')

#     parser.add_argument('image_files', type=str, default="src/compare_images/index10.png src/compare_images/index73.png "

#         , nargs='+', help='Images to compare')

#     parser.add_argument('--image_size', type=int,

#         help='Image size (height, width) in pixels.', default=160)

#     parser.add_argument('--margin', type=int,

#         help='Margin for the crop around the bounding box (height, width) in pixels.', default=44)

#     parser.add_argument('--gpu_memory_fraction', type=float,

#         help='Upper bound on the amount of GPU memory that will be used by the process.', default=0.5)

#     return parser.parse_args(argv)

if __name__ == '__main__':

    main()
"""Validate a face recognizer on the "Labeled Faces in the Wild" dataset (http://vis-www.cs.umass.edu/lfw/).

Embeddings are calculated using the pairs from http://vis-www.cs.umass.edu/lfw/pairs.txt and the ROC curve

is calculated and plotted. Both the model metagraph and the model parameters need to exist

in the same directory, and the metagraph should have the extension '.meta'.

"""

# MIT License

#

# Copyright (c) 2016 David Sandberg

#

# Permission is hereby granted, free of charge, to any person obtaining a copy

# of this software and associated documentation files (the "Software"), to deal

# in the Software without restriction, including without limitation the rights

# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell

# copies of the Software, and to permit persons to whom the Software is

# furnished to do so, subject to the following conditions:

#

# The above copyright notice and this permission notice shall be included in all

# copies or substantial portions of the Software.

#

# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR

# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,

# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE

# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER

# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,

# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE

# SOFTWARE.

from __future__ import absolute_import

from __future__ import division

from __future__ import print_function

import tensorflow as tf

import numpy as np

import argparse

import facenet

import lfw

import os

import sys

import math

from sklearn import metrics

from scipy.optimize import brentq

from scipy import interpolate

import numpy

from PIL import Image,ImageDraw

import cv2

from scipy import misc

import argparse

import align.detect_face

def detetet_face_init():

    cap = cv2.VideoCapture(0)

    print(cap.isOpened())

    classifier=cv2.CascadeClassifier("./xml/haarcascade_frontalface_alt.xml")

    count=0

    return cap,classifier,count

def detect_face_clear():

    cap.release()

    cv2.destroyAllWindows()

def load_and_align_data(image_paths, image_size, margin, gpu_memory_fraction):

    minsize = 20 # minimum size of face

    threshold = [ 0.6, 0.7, 0.7 ]  # three steps's threshold

    factor = 0.709 # scale factor

    print('Creating networks and loading parameters')

    with tf.Graph().as_default():

        gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=gpu_memory_fraction)

        sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options, log_device_placement=False))

        with sess.as_default():

            pnet, rnet, onet = align.detect_face.create_mtcnn(sess, None)

    nrof_samples = len(image_paths)

    img_list = [None] * nrof_samples

    for i in range(nrof_samples):

        img = misc.imread(os.path.expanduser(image_paths[i]))

        img_size = np.asarray(img.shape)[0:2]

        bounding_boxes, _ = align.detect_face.detect_face(img, minsize, pnet, rnet, onet, threshold, factor)

        det = np.squeeze(bounding_boxes[0,0:4])

        bb = np.zeros(4, dtype=np.int32)

        bb[0] = np.maximum(det[0]-margin/2, 0)

        bb[1] = np.maximum(det[1]-margin/2, 0)

        bb[2] = np.minimum(det[2]+margin/2, img_size[1])

        bb[3] = np.minimum(det[3]+margin/2, img_size[0])

        cropped = img[bb[1]:bb[3],bb[0]:bb[2],:]

        aligned = misc.imresize(cropped, (image_size, image_size), interp='bilinear')

        prewhitened = facenet.prewhiten(aligned)

        img_list[i] = prewhitened

    images = np.stack(img_list)

    return images

def compare_facevec(facevec1, facevec2):

    dist = np.sqrt(np.sum(np.square(np.subtract(facevec1, facevec2))))

    #print('  %1.4f  ' % dist, end='')

    return dist

def face_recognition_using_facenet():

    cap,classifier,count = detetet_face_init()

    model = "../models/20170216-091149"

    image_files = ['compare_images/index10.png', 'compare_images/index73.png']

    image_size = 160

    margin = 44

    gpu_memory_fraction = 0.5

    with tf.Graph().as_default():

        with tf.Session() as sess:

            # Load the model

            facenet.load_model(model)

            # Get input and output tensors

            images_placeholder = tf.get_default_graph().get_tensor_by_name("input:0")

            embeddings = tf.get_default_graph().get_tensor_by_name("embeddings:0")

            phase_train_placeholder = tf.get_default_graph().get_tensor_by_name("phase_train:0")

            #image_size = images_placeholder.get_shape()[1]  # For some reason this doesn't work for frozen graphs

            image_size = 160

            embedding_size = embeddings.get_shape()[1]

            index = 0

            th = 0.7

            face_recognition_tag = True

            color = (0,255,0)

            exist_face_vec = []

            face_detect_vec = []

            while count > -1:

                ret,img = cap.read()

                faceRects = classifier.detectMultiScale(img, 1.2, 2, cv2.CASCADE_SCALE_IMAGE,(20,20))

                if len(faceRects)>0:

                    for faceRect in faceRects:

                            x, y, w, h = faceRect

                            #cv2.rectangle(img, (int(x), int(y)), (int(x)+int(w), int(y)+int(h)), (0,255,0), 2,0)

                            #print "save faceimg"

                            face_win = img[int(y):int(y) + int(h), int(x):int(x) + int(w)]

                            face_detect = cv2.resize(face_win,(image_size,image_size),interpolation=cv2.INTER_CUBIC)

                            #cv2.imwrite('faceimg/index' + str(index) + '.bmp', face_win)

                            # Run forward pass to calculate embeddings

                            #print('Runnning forward pass on face detect')

                            nrof_samples = 1

                            img_list = [None] * nrof_samples

                            prewhitened = facenet.prewhiten(face_detect)

                            img_list[0] = prewhitened

                            images = np.stack(img_list)

                            if index == 10:

                                feed_dict = {images_placeholder:images, phase_train_placeholder:False }

                                exist_face_vec = sess.run(embeddings, feed_dict=feed_dict)

                            elif index > 10  and index % 10 == 0:

                                feed_dict = {images_placeholder:images, phase_train_placeholder:False }

                                face_detect_vec = sess.run(embeddings, feed_dict=feed_dict)

                                cp = compare_facevec(face_detect_vec, exist_face_vec)

                                print("index ", index, " dist ", cp)

                                if cp < th:

                                    print(True)

                                    face_recognition_tag = True

                                else:

                                    print(False)

                                    face_recognition_tag = False

                            index +=1

                            # if face_recognition_tag == True:

                            #     cv2.rectangle(img, (int(x), int(y)), (int(x)+int(w), int(y)+int(h)), (255,0,0), 2,0)

                            # else:

                            #     cv2.rectangle(img, (int(x), int(y)), (int(x)+int(w), int(y)+int(h)), (0,255,0), 2,0)

                            cv2.rectangle(img, (int(x), int(y)), (int(x)+int(w), int(y)+int(h)), (0,255,0), 2,0)

                cv2.imshow('video',img)

                key=cv2.waitKey(1)

                if key==ord('q'):

                    break

    detect_face_clear(cap)

if __name__ == '__main__':

    face_recognition_using_facenet()

python 人脸识别的更多相关文章

  1. Python人脸识别最佳教材典范，40行代码搭建人脸识别系统！

    Face Id是一款高端的人脸解锁软件,官方称:"在一百万张脸中识别出你的脸."百度.谷歌.腾讯等各大企业都花费数亿来鞭策人工智能的崛起,而实际的人脸识别技术是否有那么神奇? 绿帽 ...

  2. python人脸识别

    需要掌握知识python,opencv和机器学习一类的基础 过一段时间代码上传github,本人菜j一个,虽然是我自己谢的,也有好多不懂,或者我这就是错误方向 链接:https://pan.baidu ...

  3. 【python人脸识别】使用opencv识别图片中的人脸

    概述: OpenCV是一个基于BSD许可(开源)发行的跨平台计算机视觉库 为什么有OpenCV? 计算机视觉市场巨大而且持续增长,且这方面没有标准API,如今的计算机视觉软件大概有以下三种: 1.研究 ...

  4. Python人脸识别 + 手机推送，老板来了你就会收到短信提示

  5. 总结几个简单好用的Python人脸识别算法

    原文连接:https://mp.weixin.qq.com/s/3BgDld9hILPLCIlyysZs6Q 哈喽,大家好. 今天给大家总结几个简单.好用的人脸识别算法. 人脸识别是计算机视觉中比较常 ...

  6. OpenCV+python 人脸识别

    首先给大家推荐一本书:机器学习算法原理与编程实践 本文内容全部转载于书中,相当于一个读书笔记了吧 绪论 1992年麻省理工学院通过实验对比了基于结构特征的方法与基于模版匹配的方法,发现模版匹配的方法要 ...

  7. 简单的 Python 人脸识别实例

    案例一 导入图片 思路: 1.导入库 2.加载图片 3.创建窗口 4.显示图片 5.暂停窗口 6.关闭窗口 # 1.导入库 import cv2 # 2.加载图片 img = cv2.imread(' ...

  8. python人脸识别项目face-recognition

    该项目基于Github上面的开源项目人脸识别face-recognition,主要是对图像和视频中的人脸进行识别,在开源项目给出的例子基础上对视频人脸识别的KNN算法进行了实现. 0x1 工程项目结构 ...

  9. python 人脸识别试水（一）

    1.安装python,在这里我的版本是python 3.6 2.安装pycharm,我的版本是pycharm 2017 3.安装pip  pip 版本10 4.安装 numpy    :pip ins ...

随机推荐

  1. 黑马程序员_<<TCP>>

    ---------------------ASP.Net+Android+IOS开发..Net培训.期待与您交流! -------------------- 1. TCP/IP 1.概述 要建立连接, ...

  2. Java基础语法(第1章变量)

    今日内容介绍 1.变量 2. 运算符   变量 1.1.变量概述 前面我们已经学习了常量,接下来我们要学习变量.在Java中变量的应用比常量的应用要多很多.所以变量也是尤为重要的知识点! 什么是变量? ...

  3. 算法笔记_043:最大连续子数组和（Java）

    目录 1 问题描述 2 解决方案 2.1 蛮力枚举法 2.2 动态规划法   1 问题描述 给定一个整数数组,数组里可能有正数.负数和零.数组中连续的一个或多个整数组成一个子数组,每个子数组都有一个和 ...

  4. GridLayout with span

    Widgets can span multiple columns or rows in a grid. In the next example we illustrate this. #!/usr/ ...

  5. MVC中cshtml文件中怎样使用ViewBag进行筛选？

    @foreach (PermissionForRoleModel item in ((List<PermissionForRoleModel>)ViewBag.PermissionsFor ...

  6. CSerialPort串口类最新修正版(解决关闭死锁问题)2014-01-11

    这是一份优秀的类文件,好多的地方值得我们学习,具体在多线程,事件,自定义消息,类的封装方面等等.Remon提供的串口类网址为:http://codeguru.earthweb.com/network/ ...

  7. 如何修改浏览器默认的alert样式？

    window.alert = function(str) { var shield = document.createElement("DIV"); shield.id = &qu ...

  8. 关于python 中的 sys.argv 的使用方法

    sys.argv是获取在cmd运行python文件的时候输入的命令行参数,呈现的数据结构是列表的格式 1.用pacharm时运行时的结果是: 输出结果: 2.当我在cmd中输入指令      debu ...

  9. js 获取中文的拼音首字母

    es6 + 模块化封装 "use strict"; module.exports = { //参数,中文字符串 //返回值:拼音首字母串数组 makePy (str) { if ( ...

  10. [转]Oracle 10g如何对用户姓名，按首字母排序、查询

    首先介绍Oracle 9i新增加的一个系统自带的排序函数  1.按首字母排序  在oracle9i中新增了按照拼音.部首.笔画排序功能.设置NLS_SORT值      SCHINESE_RADICA ...