Table 1.
AlexNet 2D CNN model.
| Layer | Output shape | Param (#) |
|---|---|---|
| Conv 2d (conv 2D) | (None 32, 32, 16) | 272 |
| Batch_normalization_v1 | (None 32, 32, 16) | 64 |
| Max_pooling 2D | (None 32, 32, 16) | 0 |
| Conv2d_1 | (None 32, 32, 32) | 4640 |
| Batch_normalization_v1_1 | (None 32, 32, 32) | 128 |
| Max_pooling 2d_1 | (None 32, 32, 32) | 0 |
| Conv2d_2 | (None 32, 32, 64) | 18496 |
| Batch_normalization_v1_2 | (None 32, 32, 32) | 256 |
| Conv2d_3 | (None 32, 32, 32) | 36928 |
| Batch_normalization_v1_3 | (None 32, 32, 32) | 256 |
| Conv2d_4 (conv 2D) | (None 8, 8, 32) | 18454 |
| Batch_normalization_v1_4 | (None 8, 8, 32) | 128 |
| Max_pooling 2d_2 | (None 8, 8, 32) | 0 |
| Flatten layer | (None, 512) | 0 |
| Dense layer | (None, 200) | 102600 |
| Batch_normalization_v1_5 | (None, 200) | 800 |
| Dropout (dropout) | (None, 200) | 0 |
| Dense_1 (dense layer) | (None, 75) | 15075 |
| Batch_normalization_v1_6 | (None, 75) | 300 |
| Dropout_1 (dropout) | (None, 75) | 0 |
| Dense_2 (dense) | (None, 2) | 152 |