Skip to content

mikuchar/pytorch-adacos

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

PyTorch implementation of AdaCos

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.3%
  • Shell 0.7%