Deep Learning with PyTorch

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 5, Deep 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 ...