facenet人脸匹配(学习记录)

facenet源码码云链接

数据格式:不同人对应不同的文件夹,同一个人在一个文件夹中
文件夹xxx 文件名xxx_0001.jpg

数据预处理

数据集用lfw \64_CASIA-FaceV5\celeba

1.使用facenet源码框架中的脚本文件src/align/align_dataset_mtcnn.py来处理数据集

python src/align/align_dataset_mtcnn.py /home/wei/baidunetdiskdownload/lfw/LFW /home/wei/baidunetdiskdownload/lfw/LFW _160  --image_size 160 --margin32 --random_order --gpu_memory_fraction 0.25
#1.脚本路径 ; 2.输入图片路径 3.输出图片路径 4.图片尺寸 5.扩充 6.样本打乱 7.gpu占用量

2.用dlib库进行数据清洗

模型训练
只修改triplet loss 第454行 --data_dir设置为自己的数据集路径

python train_tripletloss.py

当epoch大概到50次 loss接近0的时候结束训练

模型测试
python src/validate_on_lfw.py data_path model_path
#1.测试脚本 2.测试图片路径 3.模型路径

模型打包
python src/freeze_graph.py model_dir output_file

图片测试
python src/compare.py imgA_path imgB_path