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

Official repo for QHack—the quantum machine learning hackathon

The implementation of FOLD-R++ algorithm

Bayesian Optimization using GPflow

Rainbow DQN implementation that outperforms the paper's results on 40% of games using 20x less data 🌈

Code for the paper "Combining Textual Features for the Detection of Hateful and Offensive Language"

Source codes of CenterTrack++ in 2021 ICME Workshop on Big Surveillance Data Processing and Analysis

A collection of semantic image segmentation models implemented in TensorFlow

Recurrent Variational Autoencoder that generates sequential data implemented with pytorch

Bayesian Image Reconstruction using Deep Generative Models

TorchXRayVision: A library of chest X-ray datasets and models.

A PyTorch-centric hybrid classical-quantum machine learning framework

Code for Paper: Self-supervised Learning of Motion Capture

Implicit Graph Neural Networks

Code for ICLR 2020 paper "VL-BERT: Pre-training of Generic Visual-Linguistic Representations".

UNet model with VGG11 encoder pre-trained on Kaggle Carvana dataset

Lyapunov-guided Deep Reinforcement Learning for Stable Online Computation Offloading in Mobile-Edge Computing Networks

Filtering variational quantum algorithms for combinatorial optimization

OCRA (Object-Centric Recurrent Attention) source code

Patch-Diffusion Code (AAAI2022)

Hand Gesture Volume Control is AIML based project which uses image processing to control the volume of your Computer.