Skip to content

Implementation of an image classification pipeline using CNN to classify the natural scenes images into different classes.

Notifications You must be signed in to change notification settings

AmeeraArif09/Multiclass-classification-with-CNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

43 Commits
 
 
 
 
 
 

Repository files navigation

CV-Assignment-3 (Multi-class classification using CNN)

Implementation of an image classification pipeline using CNN to classify the natural scenes images into different classes.

VGGnet CNN architecture is used in Keras. The different training settings, network diagrams, confusion matrics, graphs of training and validation losses and accuracies are shown in each code.

How to run the codes?

  1. Upload each Python file in a separate Colab file.
  2. Upload the test data separately. (I made a small folder of the larger set)
  3. Run all to get the results.

Methodolgies and results

1. VGGnet is built from scratch and is trained on Intel image classification dataset. Using different parameters to get best training accuracy of 96.20%

validation accuracy of 80.59%

test accuracy of 78.67%

The training and validation accuracy plot of this part is:

self_made_best_plot

The training and validation accuracy plot showing all experimentations is:

self made all

2. VGGnet is built from scratch and is trained on augmented Intel image classification dataset. Using different parameters to get best training accuracy of 97.03%

validation accuracy of 80.63%

test accuracy of 79.23%

The training and validation accuracy plot of this part is:

self made best plot image aug

The training and validation accuracy plot showing all experimentations is:

self made image aug all

3. Pre-implemented VGGnet is used on Intel image classification dataset with Transfer learning technique. Using different parameters to get best

training accuracy of 70.55%

validation accuracy of 69.80%

test accuracy of 72.33%

The training and validation accuracy plot showing best results is:

transfer learning best plot

The training and validation accuracy plot showing all experimentations is:

transfer learning all plots

4. Pre-implemented VGGnet is used on augmented Intel image classification dataset with Transfer learning technique. Using different parameters to get best

training accuracy of 76.90%

validation accuracy of 74.23%

test accuracy of 72.48%

The training and validation accuracy plot using augmentation and showing best results is:

SGD l,a plot crop

The training and validation accuracy plot showing all experimentations is:

all l,a crop aug

The result of trained model is visualized in following figure which shows that the test labels have been correctly predicted and hence our model is accurately classifying images of dataset.

inference

About

Implementation of an image classification pipeline using CNN to classify the natural scenes images into different classes.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages