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

Fast and scalable uncertainty quantification for neural molecular property prediction, accelerated optimization, and guided virtual screening.

OpenMMLab 3D Human Parametric Model Toolbox and Benchmark

Multi Camera Calibration

Effect of Different Encodings and Distance Functions on Quantum Instance-based Classifiers

A PyTorch implementation of QANet.

PyTorch Code for NeurIPS 2021 paper Anti-Backdoor Learning: Training Clean Models on Poisoned Data.

Theory-inspired Parameter Control Benchmarks for Dynamic Algorithm Configuration

Using python and scikit-learn to make stock predictions

PyTorch implementation of the supervised learning experiments from the paper Model-Agnostic Meta-Learning (MAML)

JudeasRx - graphical app for doing personalized causal medicine using the methods invented by Judea Pearl et al.

An API-first distributed deployment system of deep learning models using timeseries data to analyze and predict systems behaviour

Official page of Struct-MDC (RA-L'22 with IROS'22 option); Depth completion from Visual-SLAM using point & line features

Evaluation Pipeline for our ECCV2020: Journey Towards Tiny Perceptual Super-Resolution.

FedCV: A Federated Learning Framework for Diverse Computer Vision Tasks

This is the official pytorch implementation for our ICCV 2021 paper "TRAR: Routing the Attention Spans in Transformers for Visual Question Answering" on VQA Task

Implementation of Bidirectional Recurrent Independent Mechanisms (Learning to Combine Top-Down and Bottom-Up Signals in Recurrent Neural Networks with Attention over Modules)

Implementation of the CVPR 2021 paper "Online Multiple Object Tracking with Cross-Task Synergy"

Adaptive Attention Span for Reinforcement Learning

This repository contains implementations of all Machine Learning Algorithms from scratch in Python. Mathematics required for ML and many projects have also been included.

Python periodic table module