Table 5 EfficientNet-B0 model architecture with layer information (input and output sizes).
Layer | Size in | Size out |
|---|---|---|
Top_activation (layer 236 in EfficientNet-B0) | (None, 7, 7, 1280) | (None, 7, 7, 1280) |
Average_pooling2d | (None, 7, 7, 1280) | (None, 1, 1, 1280) |
Dense_1 | (None, 1, 1, 1280) | 256 |
Dense_2 | 256 | 128 |
Dense_3 | 128 | 128 |
Concatenate | (Dense_1, Dense_2, Dense_3) | 512 |
Softmax | 512 | 4 |
