Distributed Arcface Training in Pytorch

This is a deep learning library that makes face recognition efficient, and effective, which can train tens of millions identity on a single server.

Requirements

Install pytorch (torch>=1.6.0), our doc for install.md.
pip install -r requirements.txt.
Download the dataset from https://github.com/deepinsight/insightface/tree/master/recognition/datasets .

How to Training

To train a model, run train.py with the path to the configs:

python train.py configs/webface_mbf

Model Zoo

inference

python3 inference_compare.py --img1 input/sajjad_0.jpg --img2 input/sajjad_1.jpg

Test

We tested many versions of PyTorch. Please create an issue if you are having trouble.

torch 1.6.0
torch 1.7.1
torch 1.8.0
torch 1.9.0

Citation

@inproceedings{deng2019arcface,
  title={Arcface: Additive angular margin loss for deep face recognition},
  author={Deng, Jiankang and Guo, Jia and Xue, Niannan and Zafeiriou, Stefanos},
  booktitle={Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition},
  pages={4690--4699},
  year={2019}
}
@inproceedings{an2020partical_fc,
  title={Partial FC: Training 10 Million Identities on a Single Machine},
  author={An, Xiang and Zhu, Xuhan and Xiao, Yang and Wu, Lan and Zhang, Ming and Gao, Yuan and Qin, Bin and
  Zhang, Debing and Fu Ying},
  booktitle={Arxiv 2010.05222},
  year={2020}
}

A deep learning library that makes face recognition efficient and effective

Related tags

Overview

Distributed Arcface Training in Pytorch

Requirements

How to Training

Model Zoo

inference

Test

Citation

Owner

Sajjad Aemmi

Scalable, event-driven, deep-learning-friendly backtesting library

An intuitive library to extract features from time series

[CVPR2021] UAV-Human: A Large Benchmark for Human Behavior Understanding with Unmanned Aerial Vehicles

🍀 Pytorch implementation of various Attention Mechanisms, MLP, Re-parameter, Convolution, which is helpful to further understand papers.⭐⭐⭐

This repository contains datasets and baselines for benchmarking Chinese text recognition.

End-To-End Optimization of LiDAR Beam Configuration

Multi-Scale Aligned Distillation for Low-Resolution Detection (CVPR2021)

PyTorch implementation of "Transparency by Design: Closing the Gap Between Performance and Interpretability in Visual Reasoning"

Styled Augmented Translation

The official GitHub repository for the Argoverse 2 dataset.

GAN-STEM-Conv2MultiSlice - Exploring Generative Adversarial Networks for Image-to-Image Translation in STEM Simulation

EncT5: Fine-tuning T5 Encoder for Non-autoregressive Tasks

Source code of NeurIPS 2021 Paper ''Be Confident! Towards Trustworthy Graph Neural Networks via Confidence Calibration''

Simple-System-Convert--C--F - Simple System Convert With Python

Generative Handwriting using LSTM Mixture Density Network with TensorFlow

[ICLR 2021 Spotlight Oral] "Undistillable: Making A Nasty Teacher That CANNOT teach students", Haoyu Ma, Tianlong Chen, Ting-Kuei Hu, Chenyu You, Xiaohui Xie, Zhangyang Wang

Skyformer: Remodel Self-Attention with Gaussian Kernel and Nystr\"om Method (NeurIPS 2021)

Speeding-Up Back-Propagation in DNN: Approximate Outer Product with Memory

The implementation of our CIKM 2021 paper titled as: "Cross-Market Product Recommendation"

"NAS-Bench-301 and the Case for Surrogate Benchmarks for Neural Architecture Search".