Skip to content

MxNet Gluon Implementation of Center Loss: A Discriminative Feature Learning Approach for Deep Face Recognition

Notifications You must be signed in to change notification settings

ShownX/mxnet-center-loss

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

MxNet implementation of the paper: A Discriminative Feature Learning Approach for Deep Face Recognition

Requirements

pip install -r requirements.txt

Training

  1. Train with original softmax
$ python main.py --train --prefix=softmax
  1. Train with softmax + center loss
$ python main.py --train --center_loss --prefix=center-loss

Test

  1. Test with original softmax
$ python main.py --test --prefix=softmax
  1. Test with softmax + center loss
$ python main.py --test --prefix=center-loss

Image

Comparison Accuracy curve:

Softmax

Training:

Testing:

Softmax + Center Loss

Training:

Testing:

About

MxNet Gluon Implementation of Center Loss: A Discriminative Feature Learning Approach for Deep Face Recognition

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages