MuggleWang / CosFace_pytorch

Pytorch implementation of CosFace
MIT License
242 stars 52 forks source link
cosface pytorch

CosFace_pytorch

Pytorch implementation of CosFace




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%