Most Basic Code for and Image Recognition System using cifar-10 dataset

puru chaudhary
2 min readOct 9, 2021

Cifar-10 dataset contains Images of 10 Class Categories and looks somewhat like this

Cifar-10 Dataset with 10 Classes

Now the most basic model for an image recognition system that can be created using this dataset will predict the classes to which that object belongs along with the likelihood.

Prior knowledge of Machine learning and python is required to understand the below code

At the end of the image Epoch 1/30 represents the first pass for training the model through the whole dataset as we have specified epoch value as 30 there will be 30 total passes of training taking place through the same training dataset

As we train the model we store the weight of nodes in an h5 file named model_weights. Next, to predict the results we first convert the image to a 32* 32-pixel image as that is the input size of this model.

--

--