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

MDMM - Learning multi-domain multi-modality I2I translation

Code release for "BoxeR: Box-Attention for 2D and 3D Transformers"

Code for Low-Cost Algorithmic Recourse for Users With Uncertain Cost Functions

Computationally efficient algorithm that identifies boundary points of a point cloud.

Learning a mapping from images to psychological similarity spaces with neural networks.

A high performance implementation of HDBSCAN clustering.

ruptures: change point detection in Python

TigerLily: Finding drug interactions in silico with the Graph.

VQMIVC - Vector Quantization and Mutual Information-Based Unsupervised Speech Representation Disentanglement for One-shot Voice Conversion

The code of Zero-shot learning for low-light image enhancement based on dual iteration

Official Repository of NeurIPS2021 paper: PTR

BasicRL: easy and fundamental codes for deep reinforcement learning。It is an improvement on rainbow-is-all-you-need and OpenAI Spinning Up.

A 3D Dense mapping backend library of SLAM based on taichi-Lang designed for the aerial swarm.

Speech Separation Using an Asynchronous Fully Recurrent Convolutional Neural Network

HNECV: Heterogeneous Network Embedding via Cloud model and Variational inference

Depth image based mouse cursor visual haptic

Try out deep learning models online on Google Colab

Just playing with getting CLIP Guided Diffusion running locally, rather than having to use colab.

DenseCLIP: Language-Guided Dense Prediction with Context-Aware Prompting

DISTIL: Deep dIverSified inTeractIve Learning.