| | Layers | Output size | Parameter |
| | Conv2d | (None, 78, 78, 32) | 320 | | Activation (ReLU) | (None, 78, 78, 32) | 0 | | max_pooling2d | (None, 39, 39, 32) | 0 | | Conv2d_1 | (None, 37, 37, 64) | 18, 496 | | Activation_1 (ReLU) | (None, 37, 37, 64) | 0 | | max_pooling2d_1 | (None, 18, 18, 64) | 0 | | Conv2d_2 | (None, 16, 16, 128) | 73, 856 | | Activation_2 (ReLU) | (None, 16, 16, 128) | 0 | | max_pooling2d_2 | (None, 8, 8, 128) | 0 | | Flatten | (None, 8192) | 0 | | Dense | (None, 256) | 2, 097, 408 | | Activation_3 (ReLU) | (None, 256) | 0 | | Dropout | (None, 256) dropout = 0.5 | 0 | | Dense_1 | (None, 32) | 8, 224 |
|
|