_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
input_1 (InputLayer) [(None, 224, 224, 3)] 0
conv2d (Conv2D) (None, 54, 54, 96) 34944
batch_normalization (BatchN (None, 54, 54, 96) 384
ormalization)
max_pooling2d (MaxPooling2D (None, 26, 26, 96) 0
)
conv2d_1 (Conv2D) (None, 26, 26, 256) 614656
batch_normalization_1 (Batc (None, 26, 26, 256) 1024
hNormalization)
max_pooling2d_1 (MaxPooling (None, 12, 12, 256) 0
2D)
conv2d_2 (Conv2D) (None, 12, 12, 384) 885120
batch_normalization_2 (Batc (None, 12, 12, 384) 1536
hNormalization)
conv2d_3 (Conv2D) (None, 12, 12, 384) 1327488
batch_normalization_3 (Batc (None, 12, 12, 384) 1536
hNormalization)
conv2d_4 (Conv2D) (None, 12, 12, 256) 884992
batch_normalization_4 (Batc (None, 12, 12, 256) 1024
hNormalization)
max_pooling2d_2 (MaxPooling (None, 5, 5, 256) 0
2D)
=================================================================
_________________________________________________________________
Layer (type) Output Shape Param #
=================================================================
flatten (Flatten) (None, 6400) 0
dense (Dense) (None, 4096) 26218496
dropout (Dropout) (None, 4096) 0
dense_1 (Dense) (None, 4096) 16781312
dropout_1 (Dropout) (None, 4096) 0
dense_2 (Dense) (None, 5) 20485
=================================================================
Total params: 46,772,997
Trainable params: 46,770,245
Non-trainable params: 2,752
_________________________________________________________________