7/28/22, 11:15 AM Build_CIFAR_10_Dataset_classifier.
ipynb - Colaboratory
Buid CIFAR-10 Dataset classifier Using Deep learning and
Deploy it Using Streamlit
1- Import Libraries
import tensorflow as tf
from keras import datasets,layers,models
import matplotlib.pyplot as plt
import numpy as np
2-Import some Important Functions From helper.py
# Import Helper Function
!wget https://raw.githubusercontent.com/mrdbourke/tensorflow-deep-learning/main/extras/helper_
--2022-07-27 17:08:47-- https://raw.githubusercontent.com/mrdbourke/tensorflow-deep-lea
Resolving raw.githubusercontent.com (raw.githubusercontent.com)... 185.199.108.133, 185
Connecting to raw.githubusercontent.com (raw.githubusercontent.com)|185.199.108.133|:443
HTTP request sent, awaiting response... 200 OK
Length: 10246 (10K) [text/plain]
Saving to: ‘helper_functions.py’
helper_functions.py 100%[===================>] 10.01K --.-KB/s in 0s
2022-07-27 17:08:47 (115 MB/s) - ‘helper_functions.py’ saved [10246/10246]
3-Load Dataset from Tensorflow Datasets
(X_train,y_train),(X_test,y_test)=datasets.cifar10.load_data()
Downloading data from https://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz
170500096/170498071 [==============================] - 4s 0us/step
170508288/170498071 [==============================] - 4s 0us/step
X_train.shape
https://colab.research.google.com/drive/1cNoTzQfSU9LMJoQ-lMRqohaVv1HeyOTi 1/8
7/28/22, 11:15 AM Build_CIFAR_10_Dataset_classifier.ipynb - Colaboratory
(50000, 32, 32, 3)
X_test.shape
(10000, 32, 32, 3)
y_train.shape
(50000, 1)
X_train.ndim, y_train.ndim, X_test.ndim, y_test.ndim
(4, 2, 4, 2)
AS we Have Seen Our y_train and y_test is 2D but for Classification, We want 1D.So we convert
it into 1D Array
y_train=y_train.reshape(-1,)
y_train[:5]
array([6, 9, 9, 4, 1], dtype=uint8)
y_test=y_test.reshape(-1,)
y_train.ndim, y_test.ndim
(1, 1)
Now We have Seen Our y_train and y_test is 1D Array
4- Plot some Images
classes=['airplane','automobile','bird','cat','deer','dog','frog','horse','ship','truck']
def plot_sample_images(X,y,index):
plt.figure(figsize=(15,2))
plt.imshow(X[index])
plt.xlabel(classes[y[index]])
plot_sample_images(X_train,y_train, 3)
https://colab.research.google.com/drive/1cNoTzQfSU9LMJoQ-lMRqohaVv1HeyOTi 2/8
7/28/22, 11:15 AM Build_CIFAR_10_Dataset_classifier.ipynb - Colaboratory
plot_sample_images(X_train,y_train, 7)
plot_sample_images(X_train,y_train, 9)
5- Normalize The Data
X_train=X_train/255.0
X_test=X_test/255.0
X_train[:5]
array([[[[0.23137255, 0.24313725, 0.24705882],
[0.16862745, 0.18039216, 0.17647059],
[0.19607843, 0.18823529, 0.16862745],
...,
[0.61960784, 0.51764706, 0.42352941],
[0.59607843, 0.49019608, 0.4 ],
[0.58039216, 0.48627451, 0.40392157]],
[[0.0627451 , 0.07843137, 0.07843137],
[0. , 0. , 0. ],
https://colab.research.google.com/drive/1cNoTzQfSU9LMJoQ-lMRqohaVv1HeyOTi 3/8
7/28/22, 11:15 AM Build_CIFAR_10_Dataset_classifier.ipynb - Colaboratory
[0.07058824, 0.03137255, 0. ],
...,
[0.48235294, 0.34509804, 0.21568627],
[0.46666667, 0.3254902 , 0.19607843],
[0.47843137, 0.34117647, 0.22352941]],
[[0.09803922, 0.09411765, 0.08235294],
[0.0627451 , 0.02745098, 0. ],
[0.19215686, 0.10588235, 0.03137255],
...,
[0.4627451 , 0.32941176, 0.19607843],
[0.47058824, 0.32941176, 0.19607843],
[0.42745098, 0.28627451, 0.16470588]],
...,
[[0.81568627, 0.66666667, 0.37647059],
[0.78823529, 0.6 , 0.13333333],
[0.77647059, 0.63137255, 0.10196078],
...,
[0.62745098, 0.52156863, 0.2745098 ],
[0.21960784, 0.12156863, 0.02745098],
[0.20784314, 0.13333333, 0.07843137]],
[[0.70588235, 0.54509804, 0.37647059],
[0.67843137, 0.48235294, 0.16470588],
[0.72941176, 0.56470588, 0.11764706],
...,
[0.72156863, 0.58039216, 0.36862745],
[0.38039216, 0.24313725, 0.13333333],
[0.3254902 , 0.20784314, 0.13333333]],
[[0.69411765, 0.56470588, 0.45490196],
[0.65882353, 0.50588235, 0.36862745],
[0.70196078, 0.55686275, 0.34117647],
...,
[0.84705882, 0.72156863, 0.54901961],
[0.59215686, 0.4627451 , 0.32941176],
[0.48235294, 0.36078431, 0.28235294]]],
[[[0.60392157, 0.69411765, 0.73333333],
[0.49411765, 0.5372549 , 0.53333333],
[0.41176471, 0.40784314, 0.37254902],
...,
[0.35686275, 0.37254902, 0.27843137],
[0.34117647, 0.35294118, 0.27843137],
[0.30980392, 0.31764706, 0.2745098 ]],
6-Build CNN Model
cnn_model = models.Sequential([
layers.Conv2D(filters=128, kernel_size=(3, 3), activation='relu', input_shape=(32, 32, 3)
layers.MaxPooling2D((2, 2)),
https://colab.research.google.com/drive/1cNoTzQfSU9LMJoQ-lMRqohaVv1HeyOTi 4/8
7/28/22, 11:15 AM Build_CIFAR_10_Dataset_classifier.ipynb - Colaboratory
layers.Conv2D(filters=256, kernel_size=(3, 3), activation='relu'),
layers.MaxPooling2D((2, 2)),
layers.Conv2D(filters=512, kernel_size=(3, 3), activation='relu'),
layers.MaxPooling2D((2, 2)),
layers.Flatten(),
layers.Dense(1024, activation='relu'),
layers.Dense(512, activation='relu'),
layers.Dense(10, activation='softmax')
])
cnn_model.summary()
Model: "sequential"
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
conv2d (Conv2D) (None, 30, 30, 128) 3584
max_pooling2d (MaxPooling2D (None, 15, 15, 128) 0
)
conv2d_1 (Conv2D) (None, 13, 13, 256) 295168
max_pooling2d_1 (MaxPooling (None, 6, 6, 256) 0
2D)
conv2d_2 (Conv2D) (None, 4, 4, 512) 1180160
max_pooling2d_2 (MaxPooling (None, 2, 2, 512) 0
2D)
flatten (Flatten) (None, 2048) 0
dense (Dense) (None, 1024) 2098176
dense_1 (Dense) (None, 512) 524800
dense_2 (Dense) (None, 10) 5130
=================================================================
Total params: 4,107,018
Trainable params: 4,107,018
Non-trainable params: 0
_________________________________________________________________
7-Compile and Fitting The Model
cnn_model.compile(optimizer='adam',loss='sparse_categorical_crossentropy',metrics=['accuracy'
https://colab.research.google.com/drive/1cNoTzQfSU9LMJoQ-lMRqohaVv1HeyOTi 5/8
7/28/22, 11:15 AM Build_CIFAR_10_Dataset_classifier.ipynb - Colaboratory
history = cnn_model.fit(X_train, y_train, epochs=10)
Epoch 1/10
1563/1563 [==============================] - 23s 8ms/step - loss: 1.4946 - accuracy: 0.4
Epoch 2/10
1563/1563 [==============================] - 12s 8ms/step - loss: 1.0395 - accuracy: 0.6
Epoch 3/10
1563/1563 [==============================] - 12s 8ms/step - loss: 0.8336 - accuracy: 0.7
Epoch 4/10
1563/1563 [==============================] - 12s 8ms/step - loss: 0.6993 - accuracy: 0.7
Epoch 5/10
1563/1563 [==============================] - 12s 8ms/step - loss: 0.5811 - accuracy: 0.7
Epoch 6/10
1563/1563 [==============================] - 13s 8ms/step - loss: 0.4820 - accuracy: 0.8
Epoch 7/10
1563/1563 [==============================] - 12s 8ms/step - loss: 0.3897 - accuracy: 0.8
Epoch 8/10
1563/1563 [==============================] - 12s 8ms/step - loss: 0.3220 - accuracy: 0.8
Epoch 9/10
1563/1563 [==============================] - 12s 8ms/step - loss: 0.2576 - accuracy: 0.9
Epoch 10/10
1563/1563 [==============================] - 12s 8ms/step - loss: 0.2217 - accuracy: 0.9
evaluation = cnn_model.evaluate(X_test, y_test)
313/313 [==============================] - 1s 4ms/step - loss: 1.2631 - accuracy: 0.7110
8-Predict The Model
y_pred = cnn_model.predict(X_test)
y_pred[:5]
array([[4.75410488e-04, 4.41999509e-05, 3.88847751e-04, 9.26351130e-01,
2.17072753e-04, 7.19524994e-02, 4.02876583e-04, 2.28700719e-05,
5.30102079e-05, 9.20553575e-05],
[9.38355413e-07, 2.86453404e-03, 1.13301357e-09, 4.67079118e-11,
1.82257931e-10, 7.20077990e-12, 1.11323739e-09, 3.00298984e-13,
9.97133493e-01, 1.04363824e-06],
[4.21676248e-01, 2.35859290e-01, 2.13839617e-02, 3.56658101e-02,
8.80816299e-03, 4.05204901e-03, 2.55911681e-03, 7.73529988e-03,
9.53422934e-02, 1.66917816e-01],
[9.53714550e-01, 1.22230831e-05, 1.24507089e-04, 1.09769346e-04,
4.20756623e-05, 3.92021275e-06, 2.14843851e-07, 7.07491722e-07,
4.59649637e-02, 2.69128122e-05],
[4.85299267e-08, 4.51955834e-10, 4.74112807e-04, 1.17023697e-03,
9.97552335e-01, 6.89538911e-06, 7.96215318e-04, 1.21374299e-07,
2.51752041e-09, 3.13499837e-10]], dtype=float32)
https://colab.research.google.com/drive/1cNoTzQfSU9LMJoQ-lMRqohaVv1HeyOTi 6/8
7/28/22, 11:15 AM Build_CIFAR_10_Dataset_classifier.ipynb - Colaboratory
plot_sample_images(X_test, y_test, 1), plot_sample_images(X_test, y_test, 3), plot_sample_ima
(None, None, None, None, None)
9-Saving The Model
cnn_model.save('cifer_10_using_cnn.hdf5')
https://colab.research.google.com/drive/1cNoTzQfSU9LMJoQ-lMRqohaVv1HeyOTi 7/8
7/28/22, 11:15 AM Build_CIFAR_10_Dataset_classifier.ipynb - Colaboratory
cnn_model.save('cifer_10_using_cnn.h5')
Code Text
loaded_model =tf.keras.models.load_model("/content/cifer_10_using_cnn.hdf5")
loaded_model.evaluate(X_test, y_test)
313/313 [==============================] - 1s 4ms/step - loss: 1.2631 - accuracy: 0.7110
[1.2631244659423828, 0.7110000252723694]
https://colab.research.google.com/drive/1cNoTzQfSU9LMJoQ-lMRqohaVv1HeyOTi 8/8
