Table 3.
Layer | (type) | Output Shape | Param # |
---|---|---|---|
VGG16 | (Model) | (None, 7, 7, 512) | 14,714,688 |
flatten_1 | (Flatten) | (None, 25,088) | 0 |
dense_1 | (Dense) | (None, 1024) | 25,691,136 |
dropout_1 | (Dropout) | (None, 1024) | 0 |
dense_2 | (Dense) | (None, 2) | 2050 |
Total parameters | 40,407,874 | ||
Trainable parameters | 25,693,186 | ||
Non-trainable parameters | 14,714,688 |