Official repository for ElasticFace: Elastic Margin Loss for Deep Face Recognition
Paper on arxiv: arxiv
Model | Log file | Pretrained model | checkpoint |
---|---|---|---|
ElasticFace-Arc | log file | pretrained-mode | 295672backbone.pth |
ElasticFace-Cos | log file | pretrained-mode | 295672backbone.pth |
ElasticFace-Arc+ | log file | pretrained-mode | 295672backbone.pth |
ElasticFace-Cos+ | log file | pretrained-mode | 295672backbone.pth |
Evaluation result: See: Paper with code
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
If you use any of the code provided in this repository, please cite the following paper:
@InProceedings{Boutros_2022_CVPR,
author = {Boutros, Fadi and Damer, Naser and Kirchbuchner, Florian and Kuijper, Arjan},
title = {ElasticFace: Elastic Margin Loss for Deep Face Recognition},
booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) Workshops},
month = {June},
year = {2022},
pages = {1578-1587}
}
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