Model Used-> VGG16
File Structure-> VGG16.ipynb contains code to import VGG16, modify the last fully connected layer, train and save the model. The file Model_Tesing.ipynb contains code to import the saved model, test its performance and also test with sample data.