Skip to content

tpys/face-recognition-caffe2

Repository files navigation

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