Skip to content

pomo1144/CosFace_pytorch

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

24 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CosFace_pytorch

Pytorch implementation of CosFace


  • Deep Learning Platform: PyTorch 0.4.1
  • OS: CentOS Linux release 7.5
  • Language: Python 2.7
  • CUDA: 8.0


Result(new)

Single model trained on CAISA-WebFace achieves ~99.2% accuracy on LFW (Link: https://pan.baidu.com/s/1uOBATynzBTzZwrIKC4kcAA Password: 69e6)

Note: Pytorch 0.4 seems to be very different from 0.3, which leads me to not fully reproduce the previous results. Currently still adjusting parameters....

The initialization of the fully connected layer does not use Xavier but is more conducive to model convergence.

Result(old)

Network Hyper-parameter    Accuracy on LFW
Sphere20 s=30, m=0.35 99.08%
Sphere20 s=30, m=0.40 99.23%
LResnet50E-IR(In ArcFace paper) s=30, m=0.35 99.45%

About

Pytorch implementation of CosFace

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.1%
  • Shell 0.9%