README.md

AlexNet

Submission for coding assignment of Fatima Fellowship.

Model Details

This is a simple implementation of AlexNet in Keras for the coding assignment of Fatima Fellowship.

Model Description

• Developed by: Madhav Kumar
• Model type: Convolutional Neural Network

How to Get Started with the Model

The files contains the notebook Alexnet_without_cropping.ipynb. It contains the code that was used to train the model. In the files, "weights-improvement-07-0.87.hdf5" contain the weights of the best model during training. "logs-20230326T161429Z-001.zip" contain the logs of the training.

Training Details

Training Data

The files contain the training data images in train.zip and the data with image names and labels in the train.csv.

Training Procedure

I trained the AlexNet model on the training data on Google Colab.

Preprocessing

I used data augmentaion to increase the size of training data.

Evaluation

The model was tested on a validation data which was 20% percent of the train.zip data. The maximum accuracy obtained was 86.88%.