No description, website, or topics provided.
Switch branches/tags
Nothing to show
Clone or download
Latest commit 5fcd5ab Oct 29, 2017

README.md

Face Recognition for Caffe2

Caffe2 is a lightweight, modular, speed, so i want to do face recognition experiment with it, like light cnn, center loss, large margin softmax, Angular softmax and so on.

Reference

Files

  • Original Caffe2 library
  • Large margin softmax operator
    • caffe2/operators/lsoftmax_with_loss.h
    • caffe2/operators/lsoftmax_with_loss.cc
    • caffe2/operators/lsoftmax_with_loss.cu
    • caffe2/python/operator_test/lsoftmax_with_loss_test.py
  • face_example
    • caffe2/python/examples/sphereface_trainer.py
    • caffe2/python/models/sphereface.py
  • mnist_example
    • caffe2/python/examples/mnist_trainer.py

Result

  • mnist_example for lsoftmax

    • margin = 1

    • margin = 2

    • margin = 3

    • margin = 4

Build

Use my version of caffe2, follow origin caffe2 installation Installation