tinatianth / sphereface_pytorch

A PyTorch Implementation of SphereFace.

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

φ

References

sphereface