由于是第一次做 人脸识别,自己摸索,本来师兄是叫我先看MTCNN,不知道怎么回事蹦到了FaceNet了。不过先拿FaceNet做识别出个baseline吧。
Train a classifier on own images
因为完全不知道怎么开始,所以先按照原本流程走一边。按照LFW的训练步骤:
安装依赖项
In the below description it is assumed thatinstalled, and
repo
下载LFW数据集
wget http://vis-www.cs.umass.edu/lfw/lfw.tgz
解压到 facenet\data\lfw
处理数据(校准/对齐 将数据修剪到与预训练模型使用的数据集一致大小)
1.配置路径
export PYTHONPATH=/home/hszc/Desktop/ZHIHUICHENGSHI/version1/facenet/src
不配置得话会报错
ImportError: No module named facenet
2.处理数据
cd facenet
python src/align/align_dataset_mtcnn.py data/lfw/ data/lfw_160 --image_size 160 --margin 32 --random_order --gpu_memory_fraction 0.25
注意这里!
下载预训练模型(能翻墙就直接点击 后面上传网盘链接)
~/models/facenet/
20180402-114759
with the contents
20180402-114759.pb model-20180402-114759.ckpt-275.data-00000-of-00001 model-20180402-114759.ckpt-275.index model-20180402-114759.meta
运行测试
python src/validate_on_lfw.py data/lfw_160 /home/hszc/Desktop/ZHIHUICHENGSHI/version1/facenet/models/facenet/20180402-114759
模型路径 : 绝对路径!
参考:
https://github.com/davidsandberg/facenet/wiki/Validate-on-lfw
https://www.cnblogs.com/dyufei/p/8250484.html
文章来源: 「人脸识别」学习FaceNet(1)