做下记录,脚本如下:
对比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、素材越多训练出来的模型准确度越高,即使有素材,分类也是挺麻烦的
转载请注明博客出处: