desperado1992 / DeepBoof

Java library for running and visualizing Deep Neural Networks trained in Torch.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

DeepBoof is a Java library for running deep neural networks trained using other projects (e.g. Torch and Caffe) with a focus on processing image data. Additional tools include visualization and network training. Image processing is done using BoofCV. While it has been designed to work with Torch and Caffe it does not depend either library for its core functionality.

Installation

Gradle is used to build DeepBoof and will automatically download all dependencies.

Running

To get started classifying images simply load the Gradle project into you favorite IDE (probably IntelliJ or Eclipse) and run ExampleClassifyVggCifar10. This example downloads the network and testing data, then starts the classifier.

Alternatively, you can just build and run any example from the command-line using Gradle directly:

gradle exampleRun -Pwhich=ExampleClassifyCifar10TestSet

Future Work

DeepBoof is in an early state of development. The following are areas it needs the most improvement in:

  • Support for Caffe models (already started)
  • Native implementations of functions
  • GPU implementations of functions

About

Java library for running and visualizing Deep Neural Networks trained in Torch.


Languages

Language:Java 87.3%Language:Protocol Buffer 6.3%Language:Lua 6.2%Language:Python 0.2%Language:Shell 0.0%