Deep learning framework developed by Yangqing Jia / BVLC


Caffe is a deep learning framework developed with cleanliness, readability, and speed in mind. It was created by Yangqing Jia during his PhD at UC Berkeley, and is in active development by the Berkeley Vision and Learning Center (BVLC) and by community contributors. Caffe is released under the BSD 2-Clause license.

Check out our web image classification demo!

Why use Caffe?

Clean architecture enables rapid deployment. Networks are specified in simple config files, with no hard-coded parameters in the code. Switching between CPU and GPU is as simple as setting a flag – so models can be trained on a GPU machine, and then used on commodity clusters.

Readable & modifiable implementation fosters active development. In Caffe’s first year, it has been forked by over 600 developers on Github, and many have pushed significant changes.

Speed makes Caffe perfect for industry use. Caffe can process over 40M images per day with a single NVIDIA K40 or Titan GPU*. That’s 5 ms/image in training, and 2 ms/image in test. We believe that Caffe is the fastest CNN implementation available.

Community: Caffe already powers academic research projects, startup prototypes, and even large-scale industrial applications in vision, speech, and multimedia. There is an active discussion and support community on Github.

* When files are properly cached, and using the ILSVRC2012-winning SuperVision model. Consult performance details.



Notebook examples

Citing Caffe

Please cite Caffe in your publications if it helps your research:

  Author = {Jia, Yangqing and Shelhamer, Evan and Donahue, Jeff and Karayev, Sergey and Long, Jonathan and Girshick, Ross and Guadarrama, Sergio and Darrell, Trevor},
  Journal = {arXiv preprint arXiv:1408.5093},
  Title = {Caffe: Convolutional Architecture for Fast Feature Embedding},
  Year = {2014}

If you do publish a paper where Caffe helped your research, we encourage you to update the publications wiki. Citations are also tracked automatically by Google Scholar.


The BVLC Caffe developers would like to thank NVIDIA for GPU donation, A9 and Amazon Web Services for a research grant in support of Caffe development and reproducible research in deep learning, and BVLC PI Trevor Darrell for guidance.

The BVLC members who have contributed to Caffe are (alphabetical by first name): Eric Tzeng, Evan Shelhamer, Jeff Donahue, Jon Long, Ross Girshick, Sergey Karayev, Sergio Guadarrama, and Yangqing Jia.

The open-source community plays an important and growing role in Caffe’s development. Check out the Github project pulse for recent activity, and the contributors for a sorted list.

We sincerely appreciate your interest and contributions! If you’d like to contribute, please read the developing & contributing guide.

Yangqing would like to give a personal thanks to the NVIDIA Academic program for providing GPUs, Oriol Vinyals for discussions along the journey, and BVLC PI Trevor Darrell for advice.

Contacting us

All questions about usage, installation, code, and applications should be searched for and asked on the caffe-users mailing list.

All development discussion should be carried out at GitHub Issues.

If you have a proposal that may not be suited for public discussion and an ability to act on it, please email us directly. Requests for features, explanations, or personal help will be ignored; post such matters publicly as issues.

The core Caffe developers may be able to provide consulting services for appropriate projects.