Skip to content

Convolutional neural network to classify the CIFAR10 dataset

Notifications You must be signed in to change notification settings

nestorojeda/CIFAR-10-CNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 

Repository files navigation

CIFAR-10 Convolutional Neural Network

Implementation of two models in Keras for the classification of the CIFAR-10 dataset.

First model

It's a simple model with three Convolution+Max Pooling+Dropuout layers and two fully-connected layers. It has a validation accuracy of 78%.

Second model

The second model uses augmentation and three different types of normalization:

  • Batch normalization
  • Kernel regularization
  • Dropout

I also use a learning rate scheduler to change the lr in the middle of the training

README WIP

About

Convolutional neural network to classify the CIFAR10 dataset

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published