Skip to content

nyk510/chainer-center-loss

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

36 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Chainer Center Loss

Implementation of Center Loss in Chainer.

Requirements

  • Python 3.5.1
  • Chainer 2.0.0
  • CuPy 1.0.0 (if use GPU)
  • Matplotlib

How to train

  • with CPU and use center loss
python train_mnist.py --batchsize 32 --epoch 20 --centerloss
  • with GPU and use center loss
python train_mnist.py --batchsize 32 --epoch 20 --gpu 0 --centerloss

Visualize the result

  • MNIST, Softmax Loss + Center Loss
    The white dots denote 10 class centers of deep features.

Figure 1

  • MNIST, only Softmax Loss

Figure 2

Reference

About

Implementation of Center Loss in Chainer

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%