efps / face-identification-with-cnn-triplet-loss

Face identification with cnn+triplet-loss written by Keras.

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:

1. Face-detection-with-mobilenet-ssd
2. Face-Alignment-with-simple-cnn
3. Face-identification-with-cnn-triplet-loss

prepare data

1. Download caltech faces from Official Website , and put it into face_data folder in organize_data.py.
2. Run organize_data.py to generate train, test, dev data.
3. 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