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

Image classification using transfer learning

In this chapter, we solved an interesting dog versus cat classification problem using the transfer learning technique. We used a pre-trained VGG16 model and its weights, and subsequently we fine-tuned the training with a real-life cat versus dog dataset from Kaggle.

Once the training was completed, we saved the trained model for model persistence and later reuse. We saw that the trained model could successfully detect and differentiate between cat and dog images with very different sizes, quality, and shapes.

The trained model/classifier can be used to solve cat versus dog problems in real life. Finally, the lesson was that this similar technique with minimal efforts, can be extended and used to ...

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