ImageClassification

This project is for GDIT Challenge (Option 2): Image Classification as part of PatriotHacks hackathon
The challenge details available in https://github.com/vennergd/PatriotHack

image_classification.ipynb:

This file contains the main code, which is divided into two main parts:

• Predict image classification based on the pre-trained model: We used VGG16 and VGG19, and based on the testing, VGG16 is a better model.
• Edit the pre-trained model to predict images based on the nine classifications in the images folder: Also, the VGG16 performs better in the edited version.

Contributors:

• Wejdan Bagais
• Haotian Zheng
• Padmini
• Samapriya Dandibhotla

Resources:

• https://machinelearningmastery.com/use-pre-trained-vgg-model-classify-objects-photographs/
• https://www.learnopencv.com/keras-tutorial-fine-tuning-using-pre-trained-models/