人脸相关(1)Facenet使用

model/20180408-102900
model\20180408-102900 1.png 2.png

0.运行环境

tensorflow==1.7
scipy
scikit-learn
opencv-python
h5py
matplotlib
Pillow
requests
psutil

1.代码结构

1.1 contributed

contributed/batch_represent.py 从图片目录中生成embeddings
contributed/cluster.py 人脸图像归类
contributed/clustering.py 人脸匹配
contributed/export_embeddings.py 从图片文件夹中导出embeddings和标签(numpy数组格式)
contributed/face.py 人脸检测和识别库接口
contributed/predict.py
contributed/predict.py 实时人脸识别(获取摄像头图片,并识别)

1.2 data

data/images 测试图片文件夹
data/learning_rate_*.txt 遍历次数与学习率对照表
data/pairs.txt

1.3 align

src/align/align_dataset_mtcnn.py 执行面对齐并将面部缩略图存储在输出目录中
src/align/detect_face.py 基于多任务级联卷积神经网络的人脸检测与对准
src/align/det*.npy detect_face.py 用到的数据

1.4 generative

src/generative/models/dfc_vae.py 基于“深度特征一致变分自动编码器”的变分自动编码器
src/generative/models/dfc_vae_large.py 基于“深度特征一致变分自动编码器”的变分自动编码器(大图片 128*128)
src/generative/models/dfc_vae_resnet.py 基于“深度特征一致变分自动编码器”的变分自动编码器
src/generative/models/dfc_base.py 变分自动编码器基类,包含编码器和解码器
src/generative/calculate_attribute_vectors.py 计算属性向量
src/generative/modify_attribute.py 修改属性向量
src/generative/train_vae.py 训练变分自动编码器

1.5 models

src/models/dummy.py 虚拟模型,用于测试
src/models/inception_resnet_v1.py Inception Resnet V1 网络结构
src/models/inception_resnet_v2.py Inception Resnet V2 网络结构
src/models/squeezenet.py squeezenet 网络结构

1.6 训练

src/calculate_filtering_metrics.py 计算数据集的过滤指标并存储在.hdf文件中
src/classifier.py 使用自定义数据集训练分类器
src/compare.py 进行面部对齐并计算与嵌入图片的欧氏距离
src/decode_msceleb_dataset.py 解析msceleb数据集
src/download_and_extract.py
src/facenet.py 构建人脸识别网络
src/freeze_graph.py 冻结网络,输出模型文件
src/lfw.py 评估lfw数据集
src/train_softmax.py 使用交叉熵损失函数训练
src/train_tripletloss.py 使用三元损失函数训练
src/validate_on_lfw.py 在lfw数据集上验证
test/*_test.py 测试代码
tmp
util/plot_learning_curves.m matlab中的绘制学习曲线工具

2. align_dataset_mtcnn.py

图片预处理——运行人脸对齐程序(align\align_dataset_mtcnn.py)。

下载LFW数据集用来测试这个程序,也为后边的训练函数做好数据准备。
下载链接:http://vis-www.cs.umass.edu/lfw/。下载后解压在data文件夹中。
因为程序中神经网络使用的是谷歌的“inception resnet v1”网络模型,这个模型的输入时160160的图像,而我们下载的LFW数据集是250250限像素的图像,所以需要进行图片的预处理。

在运行时需要输入的参数:

input_dir:输入图像的文件夹(E:\facenet\data\lfw)
output_dir:输出图像的文件夹(E:\facenet\data\lfw E:\facenet\data\lfw_160)
指定裁剪后图片的大小:--image_size 160 --margin 32 --random_order(如果不指定,默认的裁剪结果是182*182像素的)

比如我的是:E:\facenet\data\lfw E:\facenet\data\lfw_160 --image_size 160 --margin 32 --random_order
E:\科研\data\lfw E:\科研\data\lfw_160 --image_size 160 --margin 32 --random_order
E:\科研\data\Picture E:\科研\data\Picture_160 --image_size 160 --margin 32 --random_order
如果在pycharm中运行,按照同样的方法配置以上参数如下:

E:\科研\data\Hypertelorism-20180710 E:\科研\data\Hypertelorism-20180710_160 --image_size 160 --margin 32 --random_order --gpu_memory_fraction=0.5

E:\PycharmProjects\facedev-server\util\data\face_image\images_evaluation E:\PycharmProjects\facedev-server\util\data\face_image_160\images_evaluation --image_size 160 --margin 32 --random_order --gpu_memory_fraction=0.5

E:\PycharmProjects\facedev-server\util\data\face_image\images_background E:\PycharmProjects\facedev-server\util\data\face_image_160\images_background --image_size 160 --margin 32 --random_order --gpu_memory_fraction=0.5

--gpu_memory_fraction=0.5默认是1,设置为1时有时会不能运行,出现could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR 错误。猜测可能是显卡不能运行了。

3.validate_on_lfw

在lfw测试的命令:

python src/validate_on_lfw.py \
~/datasets/lfw/lfw_mtcnnpy_160 \
~/models/facenet/20180402-114759 \
--distance_metric 1 \
--use_flipped_images \
--subtract_mean \
--use_fixed_image_standardization

参数含义:
a)加载模型,
b) load and parse the text file with the image pairs,
c) calculate the embeddings for all the images (as well as their horizontally flipped versions) in the test set,
d) calculate the accuracy, validation rate (@FAR=-10e-3), the Area Under Curve (AUC) and the Equal Error Rate (EER) performance measures.

3.运行结果

compare运行截图

参考资料

[1] github上的facenet代码文档结构
[2] 谷歌facenet项目代码解读(一)--align_dataset_mtcnn.py
[3] 人脸识别之facenet代码实现
[4] facenet_train_classifier.py代码注释
[5] facenet_train.py代码注释
[6] facenet使用记录
[7] FaceNet源码使用方法及其迁移学习训练自己数据集的代码修改
[8]史上最全的FaceNet源码使用方法和讲解(一)(附预训练模型下载)
[9] 史上最全的FaceNet源码使用方法和讲解(二)
[10]基于tensorflow的人脸识别技术(facenet)的测试
[11] FaceNet--Google的人脸识别
[12]FaceNet解读整理
[13] FaceNet pre-trained模型以及FaceNet源码使用方法和讲解
[14] 如何应用MTCNN和FaceNet模型实现人脸检测及识别
[15] MTCNN与facenet实现实时人脸识别
[16] 08-人脸识别-FaceNet-classify.py代码阅读(说明见注释)

官方文档

[1] Classifier training of inception resnet v1
[2] Validate on LFW

问题解决

[1] ValueError: Object arrays cannot be loaded when allow_pickle=False解决方案

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 216,651评论 6 501
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,468评论 3 392
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 162,931评论 0 353
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,218评论 1 292
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,234评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,198评论 1 299
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,084评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,926评论 0 274
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,341评论 1 311
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,563评论 2 333
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,731评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,430评论 5 343
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,036评论 3 326
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,676评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,829评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,743评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,629评论 2 354

推荐阅读更多精彩内容