| | Layer (type) | Output shape | Param | Connected to |
| | Input layer | (256, 256, 3) | 0 | ā | | Separable conv2d | (256, 256, 64) | 259 | Input | | Batch normalization | (256, 256, 64) | 256 | Separable conv2d | | Separable conv2d | (256, 256, 64) | 4736 | Batch normalization | | Batch normalization | (256, 256, 64) | 256 | Separable conv2d | | Separable conv2d | (256, 256, 64) | 4224 | Batch normalization | | Batch normalization | (256, 256, 64) | 256 | Separable conv2d | | Conv2d | (256, 256, 64) | 256 | Input | | Batch normalization | (256, 256, 64) | 256 | Conv2d | | Max pooling 2d | (256, 256, 64) | 0 | Batch normalization | | Concatenate_1 | (256, 256, 128) | 0 | Max pooling 2d & batch normalization | | Separable conv2d | (256, 256, 128) | 16640 | Concatenate_1 | | Batch normalization | (256, 256, 128) | 512 | Separable conv2d | | Separable conv2d | (256, 256, 128) | 17664 | Batch normalization | | Batch normalization | (256, 256, 128) | 512 | Separable conv2d | | Separable conv2d | (256, 256, 128) | 16640 | Batch normalization | | Batch normalization | (256, 256, 128) | 512 | Separable conv2d | | Conv 2d | (256, 256, 128) | 16512 | Concatenate_1 | | Batch normalization | (256, 256, 128) | 512 | Conv 2d | | Max pooling 2d | (256, 256, 128) | 0 | Batch normalization | | Concatenate_2 | (256, 256, 256) | 0 | Max pooling 2d & batch normalization | | Separable conv2d | (256, 256, 256) | 66048 | Concatenate_2 | | Batch normalization | (256, 256, 256) | 1024 | Separable conv2d | | Separable conv2d | (256, 256, 256) | 68096 | Batch normalization | | Batch normalization | (256, 256, 256) | 1024 | Separable conv2d | | Separable conv2d | (256, 256, 256) | 66048 | Batch normalization | | Batch normalization | (256, 256, 256) | 1024 | Separable conv2d | | Conv2d | (256, 256, 256) | 65792 | Concatenate_2 | | Batch normalization | (256, 256, 256) | 1024 | Conv2d | | Max pooling 2d | (256, 256, 256) | 0 | Batch normalization | | Concatenate_3 | (256, 256, 512) | 0 | Max pooling 2d & batch normalization | | Total params: 350,083 | Trainable params: 346,499 | Nontrainable params: 3,584 |
|
|