The Image Classification was implemented using tensorflow and pytorch ML frameworks. Transfer learning method for classification was also implemented on tensorflow.

​

Part 1: Tensorflow Image Classifier Implementation.

• This notebook contains an introduction to the CIFAR-10 dataset for image classification, and shows how to build, train, and tune a model for image classification with tensorflow from scratch with an emphasis on understanding the effect of different hyperparameters and their relationships through experimentation.

• It introduces training with GPU and checks to confirm that TensorFlow is connected to your device’s GPU.

• It introduces different ways of loading data in TensorFlow and building input data pipelines: Generate TensorFlow datasets, shuffle, batch, and prefetch for efficient data loading during training.

• Introduces the Keras image pre-processing layers.

• Building and choosing a baseline model to be improved upon.

• Building convolutional neural network models with different layers like the dropout layers, batch normalization layers, maxpooling layers, flattening layers, activation layers, convolution layers, choice of activation functions, loss functions, and optimizers.

• Discusses overfitting and underfitting; the signs of overfitting and underfitting using learning plots and how to solve them.

• Examining the effect of hyperparameters and exploring the relationship between hyperparameters through experimentation hyperparameter choice and introduction of Keras tuner for hyperparameter tuning.

​

Part 2: Pytorch  Image Classifier Implementation.

• A pytorch implementation of the final tuned model was also developed.

​

Part 3: Transfer Learning Method for Image Classification.

• Finally, the TensorFlow resnet model was finetuned to achieve a validation accuracy of over 93% on the CIFAR-10 dataset.