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

Automatic labeling, conversion of different data set formats, sample size statistics, model cascade

[ICLR'21] FedBN: Federated Learning on Non-IID Features via Local Batch Normalization

⚾🤖⚾ Automatic baseball pitching overlay in realtime

An open-source, low-cost, image-based weed detection device for fallow scenarios.

Code release for NeRF (Neural Radiance Fields)

Multi-task head pose estimation in-the-wild

Pytorch0.4.1 codes for InsightFace

Boosted CVaR Classification (NeurIPS 2021)

Malware Bypass Research using Reinforcement Learning

Interactive Visualization to empower domain experts to align ML model behaviors with their knowledge.

[NeurIPS 2021 Spotlight] Code for Learning to Compose Visual Relations

Extending JAX with custom C++ and CUDA code

PyTorch framework, for reproducing experiments from the paper Implicit Regularization in Hierarchical Tensor Factorization and Deep Convolutional Neural Networks

A high-level Python library for Quantum Natural Language Processing

Pytorch implementation of Feature Pyramid Network (FPN) for Object Detection

Public Code for NIPS submission SimiGrad: Fine-Grained Adaptive Batching for Large ScaleTraining using Gradient Similarity Measurement

PyTorch implementation of SmoothGrad: removing noise by adding noise.

基于Paddle框架的arcface复现

Pytorch implementation of our paper accepted by NeurIPS 2021 -- Revisiting Discriminator in GAN Compression: A Generator-discriminator Cooperative Compression Scheme

VarCLR: Variable Semantic Representation Pre-training via Contrastive Learning