Image Augmentation for Deep Learning using PyTorch

Now, without waiting further, let's move on to the model building part. We will apply the augmentation techniques that are discussed in this article to generate images and then use those images to train the model. We will be working on the emergency vs non-emergency vehicle classification problem. You should be familiar with the problem statement if you've gone through my previous PyTorch articles. The aim of this project is to classify the images of vehicles as emergency or non-emergency. And you guessed it – it's an image classification problem.