Description
This is a implementation of cnn+triplet-loss for face identification written by keras, which is the last step of my FaceID system. You can find another two repositories as follows:
- Face-detection-with-mobilenet-ssd
- Face-Alignment-with-simple-cnn
- Face-identification-with-cnn-triplet-loss
prepare data
- Download caltech faces from Official Website , and put it into face_data folder in organize_data.py.
- Run organize_data.py to generate train, test, dev data.
- Run load_data.py to generate train_x.npy, train_y.npy and so on.
train
Follow face_train.ipynb step by step. You can change the parameters for better performance.
to do
Compare with center loss.
References
meownoid/face-identification-tpe