Change the CAISIA_DATA_DIR and LFW_DATA_DAR in config.py
to your data path.
Train the mobilefacenet model.
**Note:** The default settings set the batch size of 512, use 2 gpus and train the model on 70 epochs. You can change the settings in `config.py`
python3 train.py
Test the model on LFW.
Note: I have tested lfw_eval.py
on the caffe model at SphereFace, it gets the same result.
python3 lfw_eval.py --resume --feature_save_dir
--resume:
path of saved model--feature_save_dir:
path to save the extracted features (must be .mat file)You can just run the lfw_eval.py
to get the result, the accuracy on LFW like this:
Fold | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | 10 | AVE(ours) | Paper(112x96) |
---|---|---|---|---|---|---|---|---|---|---|---|---|
ACC | 99.00 | 99.00 | 99.00 | 98.67 | 99.33 | 99.67 | 99.17 | 99.50 | 100.00 | 99.67 | 99.30 | 99.18 |