vgg16_input

InputLayer

input:

output:

[(None, 150, 150, 3)]

[(None, 150, 150, 3)]

vgg16

Functional

input:

output:

(None, 150, 150, 3)

(None, 4, 4, 512)

flatten

Flatten

input:

output:

(None, 4, 4, 512)

(None, 8192)

dense

Dense

input:

output:

(None, 8192)

(None, 256)

dense_1

Dense

input:

output:

(None, 256)

(None, 5)

Fig: 3.8.1.1: Architecture of VGG16