This is the official repository of the paper:
ElasticFace: Elastic Margin Loss for Deep Face Recognition
Paper on arxiv: arxiv
*** Accepted CVPR workshops 2022 ***
| Model | Log file | Pretrained model |
|---|---|---|
| ElasticFace-Arc | log file | pretrained-mode |
| ElasticFace-Cos | log file | pretrained-mode |
| ElasticFace-Arc+ | log file | pretrained-mode |
| ElasticFace-Cos+ | log file | pretrained-mode |
Evaluation result: See: Paper with code
Face recognition model training
Model training: In the paper, we employ MS1MV2 as the training dataset which can be downloaded from InsightFace (MS1M-ArcFace in DataZoo) Download MS1MV2 dataset from insightface on strictly follow the licence distribution
Unzip the dataset and place it in the data folder Set the config.output and config.loss in the config/config.py
All code has been trained and tested using Pytorch 1.7.1
Face recognition evaluation
evaluation on LFW, AgeDb-30, CPLFW, CALFW and CFP-FP:
- download the data from their offical webpages.
- alternative: The evaluation datasets are available in the training dataset package as bin file
- set the config.rec to dataset folder e.g. data/faces_emore
- set the config.val_targets for list of the evaluation dataset
- download the pretrained model from link the previous table
- set the config.output to path to pretrained model weights
- run eval/evaluation.py
- the output is test.log contains the evaluation results over all epochs
To-do
- Add evaluation script
If you use any of the code provided in this repository, please cite the following paper:
Citation
@misc{boutros2021elasticface,
title={ElasticFace: Elastic Margin Loss for Deep Face Recognition},
author={Fadi Boutros and Naser Damer and Florian Kirchbuchner and Arjan Kuijper},
year={2021},
eprint={2109.09416},
archivePrefix={arXiv},
primaryClass={cs.CV}
}
License
This project is licensed under the terms of the Attribution-NonCommercial-ShareAlike 4.0
International (CC BY-NC-SA 4.0) license.
Copyright (c) 2021 Fraunhofer Institute for Computer Graphics Research IGD Darmstadt