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

RMNA: A Neighbor Aggregation-Based Knowledge Graph Representation Learning Model Using Rule Mining

Predictive AI layer for existing databases.

Code for paper "Document-Level Argument Extraction by Conditional Generation". NAACL 21'

Uncertainty Estimation via Response Scaling for Pseudo-mask Noise Mitigation in Weakly-supervised Semantic Segmentation

Transparent Transformer Segmentation

Time Dependent DFT in Tamm-Dancoff Approximation

A 3D sparse LBM solver implemented using Taichi

This is a library for training and applying sparse fine-tunings with torch and transformers.

A wrapper around SageMaker ML Lineage Tracking extending ML Lineage to end-to-end ML lifecycles, including additional capabilities around Feature Store groups, queries, and other relevant artifacts.

Implementation for ACProp ( Momentum centering and asynchronous update for adaptive gradient methdos, NeurIPS 2021)

TensorFlow implementation of ENet

Learning Visual Words for Weakly-Supervised Semantic Segmentation

Fast Differentiable Matrix Sqrt Root

The code for MM2021 paper "Multi-Level Counterfactual Contrast for Visual Commonsense Reasoning"

Learning to Stylize Novel Views

Third party Pytorch implement of Image Processing Transformer (Pre-Trained Image Processing Transformer arXiv:2012.00364v2)

High dimensional black-box optimizer using Latent Action Monte Carlo Tree Search algorithm

An addon uses SMPL's poses and global translation to drive cartoon character in Blender.

Code for "3D Human Pose and Shape Regression with Pyramidal Mesh Alignment Feedback Loop"

TalkNet 2: Non-Autoregressive Depth-Wise Separable Convolutional Model for Speech Synthesis with Explicit Pitch and Duration Prediction.