Skip to content
forked from yule-li/CosFace

Tensorflow implementation for paper CosFace: Large Margin Cosine Loss for Deep Face Recognition

Notifications You must be signed in to change notification settings

samxuxiang/CosFace

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

18 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

This project is aimmed at implementing the CosFace described by the paper CosFace: Large Margin Cosine Loss for Deep Face Recognition. The code can be trained on CASIA-Webface and the best accuracy LFW is 98.6%. The result is lower than reported by paper(99.33%), which may be caused by sphere network implemented in tensorflow. I train the sphere network implemented in tensorflow using the softmax loss and just obtain the accuracy 95.6%, which is more lower than caffe version(97.88%).

Preprocessing

I supply the preprocessed dataset in baidu pan:CASIA-WebFace-112X96,lfw-112X96. You can download and unzip them to dir dataset.

If you want to preprocess the dataset by yourself, you can refer to sphereface.

Train

./train.sh

Test

Modify the MODEL_DIR in test.sh and run ./test/sh.

If you do not want to train your model, you can download my trained model and unzip it to models dir.

About

Tensorflow implementation for paper CosFace: Large Margin Cosine Loss for Deep Face Recognition

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 97.1%
  • Shell 2.9%