O'Reilly logo

Java Deep Learning Projects by Md. Rezaul Karim

Stay ahead with the world's most comprehensive technology and business learning platform.

With Safari, you learn the way you learn best. Get unlimited access to videos, live online training, learning paths, books, tutorials, and more.

Start Free Trial

No credit card required

Architecture choice and adoption

As mentioned earlier, we will be reusing the VGG-16 pretrained model, which is already trained with different images of cat and dog breeds from ImageNet (see the list here at http://www.image-net.org/challenges/LSVRC/2014/results#clsloc). The original VGG-16 model had 1,000 classes of images to be predicted as outlined in the following diagram:

Original VGG-16 model architecture

Fortunately, the trained model and network weights are already available on the DL4J website (see http://blob.deeplearning4j.org/models/vgg16_dl4j_inference.zip) and the size is about 500 MB.

You can manually download and restore, or ...

With Safari, you learn the way you learn best. Get unlimited access to videos, live online training, learning paths, books, interactive tutorials, and more.

Start Free Trial

No credit card required