facenet模型训练

做下记录,脚本如下:

对比
python3 src/compare.py ../models/20190327-145140/ ../faces/pyimgs/dashenlin/ytwRkvSdG1000058.png ../faces/pyimgs/dashenlin/5BdWkjzSG1000058.png


对齐
python3 src/align_dataset_mtcnn.py ../faces/lfw29/ ../faces/lfw29_align/ --image_size=160



再训练(--pretrained_model,第一次训练不需要此参数)
python3 src/train_tripletloss.py --pretrained_model=../models/20190327-145140/ --models_base_dir=../models/ --data_dir=../faces/lfw29_align/

后台执行训练代码
nohup -u python3 src/train_tripletloss.py --pretrained_model=../models/20190327-145140/ --models_base_dir=../models/ --data_dir=../faces/lfw29_align/ > train.log & 2>&1 &

备注:

使用还是挺简单的,就是要注意下模型训练的参数设置问题:

1、图片太少的时候容易报错,代码里面有公式可以自己看看怎么设置合理

2、每个人放一个文件夹,一人多图,命名规则参照官方的例子

3、素材越多训练出来的模型准确度越高,即使有素材,分类也是挺麻烦的

 转载请注明博客出处:http://www.cnblogs.com/cjh-notes/

原文地址:https://www.cnblogs.com/cjh-notes/p/10679743.html