Skip to content

twinsyssy1018/caffe_multi-label

 
 

Repository files navigation

caffe_multi-label

This is a multi-label version of Caffe which is developed by the Berkeley Vision and Learning Center (BVLC) and community contributors.

The framework is based on the 2nd release candidate of Caffe 1.0.(See here: https://github.com/BVLC/caffe/releases), since the master branch of Caffe is under severe modification right now.

caffe.proto has been modified to allow multi-label.

A simple demo on MNIST:

  1. Download mnist

./data/mnist/get_mnist.sh

  1. Convert mnist

./examples/mnist/create_mnist.sh

  1. Train the LeNet model

./examples/mnist/train_lenet.sh

About

A multi-label version of caffe

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • C++ 80.4%
  • Python 8.9%
  • Cuda 4.3%
  • CMake 3.2%
  • Protocol Buffer 1.3%
  • Makefile 0.8%
  • Other 1.1%