clcarwin / sphereface_pytorch

A PyTorch Implementation of SphereFace.

MIT License

715 stars 172 forks source link

readme

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

φ

$phi(x)=\left(-1\right)^k\cdot \cos \left(x\right)-2\cdot k$

$myphi(x)=1-\frac{x^2}{2!}+\frac{x^4}{4!}-\frac{x^6}{6!}+\frac{x^8}{8!}-\frac{x^9}{9!}$

phi

References