Skip to content

Deep learning model training and evaluation using image dataset

Notifications You must be signed in to change notification settings

jsamanta/Deep-Learning-Projects

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 

Repository files navigation

Image detection using CIFAR-10 dataset

Trained and evaluated a Deep Learning model similar to VGG architecture.

Performced several experiments to improve the accuracy if the model. Firstly, used a set of 3 convolution layers of 32 channels, here batch normalization was disabled. Then used Maxpooling layer followed by a Dropout layer. Secondly, used a set of 3 convolution layers of 64 channels, here batch normalization was enabled. This was then followed by Max pooling layer and then Drop out layer. Thirdly, used a set of 3 convolution layers of 128 channels with Batch normalization enabled, followed by Maxpooling layer and Drop out layer. Lastly, I used a flatten layer followed by Dense layer and a dropout layer.

I used Max Pooling layer at the end of each set of convolution layer followed by Drop out layers for regularization of parameters. Used ReLU as activation fucntion. I tried using different parameters for each dropout layer to get better accuracy overall.

The entire architecture flow can be found in the model summary. With this architecture I was able to get about 95% validation accuracy and 87% of test accuracy.

About

Deep learning model training and evaluation using image dataset

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published