O'Reilly logo

Deep Learning with PyTorch by Vishnu Subramanian

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 deep learning

The most important step in solving any real-world problem is to get the data. Kaggle provides a huge number of competitions on different data science problems. We will pick one of the problems that arose in 2014, which we will use to test our deep learning algorithms in this chapter and improve it in Chapter 5Deep Learning for Computer Vision, which will be on Convolution Neural Networks (CNNs) and some of the advanced techniques that we can use to improve the performance of our image recognition models. You can download the data from https://www.kaggle.com/c/dogs-vs-cats/data. The dataset contains 25,000 images of dogs and cats. Preprocessing of data and the creation of train, validation, and test ...

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