Codes for the project are posted here. train.py,predict.py - 2 conv2D layers with max pooling and MLP. vGGtrain.py,vGGtest.py - using already built vgg model and fine-tuned the last layer of vGG16 and added MLP at the end. visualizationsEx.py - for visualizing the conv and max pool layers for a given image.
Vedavikas(vic006) , Aditya(kaiser1997)
Keras MatplotLib scikitlearn numpy google collab