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

TCTrack: Temporal Contexts for Aerial Tracking (CVPR2022)

A framework to train language models to learn invariant representations.

This is a file about Unet implemented in Pytorch

Probabilistic Programming and Statistical Inference in PyTorch

Official PyTorch implementation for Generic Attention-model Explainability for Interpreting Bi-Modal and Encoder-Decoder Transformers, a novel method to visualize any Transformer-based network. Including examples for DETR, VQA.

Rule Based Classification Project

Unofficial Implementation of Oboe (SIGCOMM'18').

Deep Multi-Magnification Network for multi-class tissue segmentation of whole slide images

Differential Privacy for Heterogeneous Federated Learning : Utility & Privacy tradeoffs

NAS-Bench-x11 and the Power of Learning Curves

Relative Uncertainty Learning for Facial Expression Recognition

UniMoCo: Unsupervised, Semi-Supervised and Full-Supervised Visual Representation Learning

UniLM AI - Large-scale Self-supervised Pre-training across Tasks, Languages, and Modalities

Multiview 3D object detection on MultiviewC dataset through moft3d.

PICK: Processing Key Information Extraction from Documents using Improved Graph Learning-Convolutional Networks

Keras documentation, hosted live at keras.io

Code for Private Recommender Systems: How Can Users Build Their Own Fair Recommender Systems without Log Data? (SDM 2022)

A smart Chat bot that can help to know about corona virus and Make prediction of corona using X-ray.

An Unsupervised Graph-based Toolbox for Fraud Detection

Source code, data, and evaluation details for “Cross-Lingual Citations in English Papers: A Large-Scale Analysis of Prevalence, Formation, and Ramifications”