Skip to content

Latest commit

 

History

History
34 lines (29 loc) · 968 Bytes

README.md

File metadata and controls

34 lines (29 loc) · 968 Bytes

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