Skip to content

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

Notifications You must be signed in to change notification settings

lessthanoptimal/DeepBoof

Repository files navigation

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.

Resources

Stars

Watchers

Forks

Packages

No packages published

Languages