GuoWenzhe / sphereface_pytorch

A PyTorch Implementation of SphereFace.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About

A PyTorch Implementation of SphereFace.

License:MIT License


Languages

Language:Python 100.0%