Skip to content

Fine tuning the VGG16 deep convolutional neural network to classify images from the CIFAR-10 dataset. Accuracy of ~0.70 achieved with ~3000 training images.

Notifications You must be signed in to change notification settings

subrapak/Image-Classification-Using-Keras

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 

Repository files navigation

Image-Classification-Using-Keras

Fine tuning the VGG16 deep convolutional neural network to classify images from the CIFAR-10 dataset.

Refer to vgg16_for_CIFAR10.ipynb for step by step explanations and discussion.

Description

This was a personal project, to familiarise myself with Keras, Tensorflow, Google Colab and image processing in general. I initially began working with Jupyter Notebooks and created a fairly shallow network for simple data classification - however, it was too simple to classify images. Therefore, using help from video tutorials online I learnt how to fine tune and manipulate pre-existing deep networks, and set myself the challenge of classifying the CIFAR-10 dataset with a modified VGG16 model.

Results

The modified networked managed an accuracy of just over 70%, and with better hardware and memory storage, the model should perform even better.

About

Fine tuning the VGG16 deep convolutional neural network to classify images from the CIFAR-10 dataset. Accuracy of ~0.70 achieved with ~3000 training images.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published