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

Code I use to automatically update my videos' metadata on YouTube

Code for "On the Effects of Batch and Weight Normalization in Generative Adversarial Networks"

Methods to get the probability of a changepoint in a time series.

Image-Scaling Attacks and Defenses

PyTorch Implementation of Fully Convolutional Networks. (Training code to reproduce the original result is available.)

Dataset VSD4K includes 6 popular categories: game, sport, dance, vlog, interview and city.

Implementation of Convolutional LSTM in PyTorch.

A robust camera and Lidar fusion based velocity estimator to undistort the pointcloud.

Auto-Encoding Score Distribution Regression for Action Quality Assessment

PyTorch implementation of SQN based on CloserLook3D's encoder

Pytorch Implementation for (STANet+ and STANet)

NeurIPS 2021, "Fine Samples for Learning with Noisy Labels"

Efficient semidefinite bounds for multi-label discrete graphical models.

SE-MSCNN: A Lightweight Multi-scaled Fusion Network for Sleep Apnea Detection Using Single-Lead ECG Signals

Self-supervised Multi-modal Hybrid Fusion Network for Brain Tumor Segmentation

PyTorch source code for Distilling Knowledge by Mimicking Features

This program will stylize your photos with fast neural style transfer.

Joint Discriminative and Generative Learning for Person Re-identification. CVPR'19 (Oral)

Depth-Aware Video Frame Interpolation (CVPR 2019)

Code for paper "A Critical Assessment of State-of-the-Art in Entity Alignment" (https://arxiv.org/abs/2010.16314)