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

Developing an image classifier using transfer learning

In the next section, we will see how to distinguish between dogs and cats based on their raw images. We will also see how to implement our first CNN model to deal with the raw and color image having three channels.

This project is highly inspired (but extended significantly) by the "Java Image Cat&Dog Recognition with Deep Neural Networks" article by Klevis Ramo (http://ramok.tech/). 

The code folder has three packages with a few Java files in each. Their functionalities are outlined as follows:

  • com.packt.JavaDL.DogvCatClassification.Train:
    • TrainCatvsDogVG16.java: It is used to train the network and the trained model is saved to a user specific location. Finally, it prints the results. ...

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