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/