资源算法mxnet-center-loss

mxnet-center-loss

2020-04-03 | |  92 |   0 |   0

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:

图片.png

Softmax

Training:

softmax-train.gif

Testing:

softmax-test.gif

Softmax + Center Loss

Training:

center-train.gif

Testing:

center-test.gif

上一篇:Keras-MNIST-center-loss-with-visualization

下一篇:MNIST-center-loss

用户评价
全部评价

热门资源

  • DuReader_QANet_BiDAF

    Machine Reading Comprehension on DuReader Usin...

  • ETD_cataloguing_a...

    ETD catalouging project using allennlp

  • allennlp_extras

    allennlp_extras Some utilities build on top of...

  • allennlp-dureader

    An Apache 2.0 NLP research library, built on Py...

  • honk-honk-motherf...

    honk-honk-motherfucker