资源分类
》 资源 》算法》chainer-center-loss

chainer-center-loss

2020-04-03 | |  45 |   0 |   0

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




上一篇：Center_Loss_in_MXNet

下一篇：center_loss_classification

用户评价
全部评价

热门资源

  • seetafaceJNI

    项目介绍 基于中科院seetaface2进行封装的JAVA...

  • spark-corenlp

    This package wraps Stanford CoreNLP annotators ...

  • Keras-ResNeXt

    Keras ResNeXt Implementation of ResNeXt models...

  • capsnet-with-caps...

    CapsNet with capsule-wise convolution Project ...

  • shih-styletransfer

    shih-styletransfer Code from Style Transfer ...