/pytorch-adacos

PyTorch implementation of AdaCos

Primary LanguagePythonMIT LicenseMIT

PyTorch implementation of AdaCos, ArcFace, CosFace, and SphereFace

This repository contains code for AdaCos, ArcFace, CosFace, and SphereFace based on AdaCos: Adaptively Scaling Cosine Logits for Effectively Learning Deep Face Representations implemented in PyTorch.

TODO

  • Omniglot
  • Results of Omniglot experiments
  • Train on WebFace and test on LFW

Requirements

  • Python 3.6
  • PyTorch 1.0

Training

MNIST

python mnist_train.py --metric adacos

Omniglot

cd omniglot
. download.sh
cd ..
python omniglot_train.py --metric adacos

Results

Omniglot

Method acc@1 acc@5
SphereFace 89.66 98.48
CosFace 89.68 98.23
ArcFace 89.54 98.48
AdaCos 90.06 98.55