Added contrastive loss layer, associated tests, and a siamese network example using...
authorNick Carlevaris-Bianco <carlevar@umich.edu>
Thu, 21 Aug 2014 13:32:10 +0000 (09:32 -0400)
committerNick Carlevaris-Bianco <carlevar@umich.edu>
Mon, 8 Sep 2014 20:14:58 +0000 (16:14 -0400)
commitd149c9a98d783eb64cefbbfeb02c6859dd81cfa6
tree368de63a44ff14709ba87b5cea09631f5aae5639
parentfc921bf9d6959f06705b7ea52b2e28733b3fdd42
Added contrastive loss layer, associated tests, and a siamese network example using shared weights and the contrastive loss.
14 files changed:
examples/siamese/convert_mnist_siamese_data.cpp [new file with mode: 0644]
examples/siamese/create_mnist_siamese.sh [new file with mode: 0755]
examples/siamese/mnist_siamese.ipynb [new file with mode: 0644]
examples/siamese/mnist_siamese.prototxt [new file with mode: 0644]
examples/siamese/mnist_siamese_solver.prototxt [new file with mode: 0644]
examples/siamese/mnist_siamese_train_test.prototxt [new file with mode: 0644]
examples/siamese/readme.md [new file with mode: 0644]
examples/siamese/train_mnist_siamese.sh [new file with mode: 0755]
include/caffe/loss_layers.hpp
src/caffe/layer_factory.cpp
src/caffe/layers/contrastive_loss_layer.cpp [new file with mode: 0644]
src/caffe/layers/contrastive_loss_layer.cu [new file with mode: 0644]
src/caffe/proto/caffe.proto
src/caffe/test/test_contrastive_loss_layer.cpp [new file with mode: 0644]