ShownX / Mxnet Center Loss
MxNet Gluon Implementation of Center Loss: A Discriminative Feature Learning Approach for Deep Face Recognition
Stars: ✭ 35
Programming Languages
python
139335 projects - #7 most used programming language
MxNet implementation of the paper: A Discriminative Feature Learning Approach for Deep Face Recognition
Requirements
pip install -r requirements.txt
Training
- Train with original softmax
$ python main.py --train --prefix=softmax
- Train with softmax + center loss
$ python main.py --train --center_loss --prefix=center-loss
Test
- Test with original softmax
$ python main.py --test --prefix=softmax
- Test with softmax + center loss
$ python main.py --test --prefix=center-loss
Image
Comparison Accuracy curve:
Softmax
Training:
Testing:
Softmax + Center Loss
Training:
Testing:
Note that the project description data, including the texts, logos, images, and/or trademarks,
for each open source project belongs to its rightful owner.
If you wish to add or remove any projects, please contact us at [email protected].