Skip to content

strin/caffe-opencl

 
 

Repository files navigation

Caffe

Build Status License

Notes

This is a fork of caffe that is able to run on Android with OpenCL. So it can utilize multiple devices on phone such as CPU, GPUs, etc.

Original ReadME

Caffe is a deep learning framework made with expression, speed, and modularity in mind. It is developed by the Berkeley Vision and Learning Center (BVLC) and community contributors.

Check out the project site for all the details like

and step-by-step examples.

Join the chat at https://gitter.im/BVLC/caffe

Please join the caffe-users group or gitter chat to ask questions and talk about methods and models. Framework development discussions and thorough bug reports are collected on Issues.

Happy brewing!

License and Citation

Caffe is released under the BSD 2-Clause license. The BVLC reference models are released for unrestricted use.

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

@article{jia2014caffe,
  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}
}

Additional Notes

This fork of Caffe contains an OpenCL backend and additional layers for fast image segmentation. This work is partially supported by:

  • AMD
  • HHMI Janelia
  • UZH, INI
  • ETH Zurich

For a C++ frontend and models to use for image segmentation with this fork, see:

OpenCL Backend

The backend is supposed to work with all vendors. Note however there may be problems with libOpenCL.so provided by nVidia. It is therefore recommended to install another OpenCL implementation after installing nVidia drivers. Possibilities are:

  • Intel OpenCL, recommended if you have an Intel CPU along the nVidia GPU.
  • AMD APP SDK (OpenCL), recommended if you have an AMD GPU or CPU.

Technical Report

Available on arXiv: http://arxiv.org/abs/1509.03371

About

Deep learning with Caffe on phones, with OpenCL support for CPU and GPU devices.

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • C++ 77.5%
  • Cuda 8.1%
  • Python 6.5%
  • CMake 2.6%
  • C 2.2%
  • Protocol Buffer 1.2%
  • Other 1.9%