1.首先下载facenet源码:https://github.com/davidsandberg/facenet
2.运行facenet
用pycharm打开,下载好tensorflow1.7 版本,对应的python版本为3.6,此处有个坑,安装tensoorflow1.7.0时会出现numpy和其他库版本不一致的问题,需要按照提示的逐步重新安装。
安装库:conda install 库名称==版本
删除库:conda uninstall 库名称
3.下载flw数据集
链接: http://vis-www.cs.umass.edu/lfw/
下载后,解压到data\lfw_data\lfw,没有文件夹则自己创建
4.对照片进行预处理
此处下载的人脸照片为250250像素,需要改成160160像素,
在pycharm中打开align_dataset_mtcnn.py文件,配置parameters:
下载数据集 存放处理后的数据集 处理设置(根据路径创建文件)
D:\yanyi\project_process\facenet-master\src\data\lfw_data\lfw D:\yanyi\project_process\facenet-master\src\data\lfw_data\lfw_160 --image_size 160 --margin 32 --random_order --gpu_memory_fraction 0.25
处理完照片后,可以在lfw_160文件中看到处理后的文件
在data文件下还会生成一个paris.txt文件
5.对模型进行测试
方式一: 下载facenet训练模型
点击下载链接
此处使用的是20180408-102900文件下的模型,将此文件复制到 facenet\src\models 目录下。
在此我首先使用pycharm添加参数:
数据集所在路径 模型所在路径
如 D:\yanyi\project_process\facenet-master\src\data\lfw_data\lfw_160 D:\yanyi\project_process\facenet-master\src\models\20180408-102900
将参数添加到parameters,运行 validate_on_lfw.py 后发现
一直出现缺少 data\paris.txt文件错误,目前还没找到解决办法,但是使用cmd运行后,成功运行!
cd 到src文件下:
python validate_on_lfw.py D:\yanyi\project_process\facenet-master\src\data\lfw_data\lfw_160 D:\yanyi\project_process\facenet-master\src\models\20180408-102900
识别精度达到0.97,测试成功
出现一些错误,目前还不懂