/sphereface_pytorch

A PyTorch Implementation of SphereFace.

Primary LanguagePythonMIT LicenseMIT

SphereFace

A PyTorch Implementation of SphereFace. The code can be trained on CASIA-Webface and the best accuracy on LFW is 99.22%.

SphereFace: Deep Hypersphere Embedding for Face Recognition

Train

python train.py

Test

# lfw.tgz to lfw.zip
tar zxf lfw.tgz; cd lfw; zip -r ../lfw.zip *; cd ..

# lfw evaluation
python lfw_eval.py --model model/sphere20a_20171020.pth

Pre-trained models

Model name LFW accuracy Training dataset
20171020 0.9922 CASIA-WebFace

φ

equation

equation

phi

References

sphereface