Advances, Systems and Applications
Table 1 Layer and parameter settings
From: Feature-enhanced fusion of U-NET-based improved brain tumor images segmentation
Part | Layer Type | Layer Name | Output Shape | Number of Parameters |
|---|---|---|---|---|
Encoder | Conv2D | conv2d | (None, 240, 240, 64) | 2368 |
Encoder | Conv2D | conv2d_1 | (None, 240, 240, 64) | 36928 |
Encoder | MaxPooling2D | max_pooling2d | (None, 120, 120, 64) | 0 |
Encoder | Dropout | dropout_1 | (None, 120, 120, 64) | 0 |
Encoder | Conv2D | conv2d_2 | (None, 120, 120, 128) | 73856 |
Encoder | Conv2D | conv2d_3 | (None, 120, 120, 128) | 147584 |
Encoder | MaxPooling2D | max_pooling2d_1 | (None, 60, 60, 128) | 0 |
Encoder | Dropout | dropout_2 | (None, 60, 60, 128) | 0 |
Encoder | Conv2D | conv2d_4 | (None, 60, 60, 256) | 295168 |
Encoder | Conv2D | conv2d_5 | (None, 60, 60, 256) | 590080 |
Transformer | Dropout | dropout_3 | (None, 30, 30, 256) | 0 |
Transformer | MultiHeadAttention | multi_head_attention | (None, 30, 30, 256) | 263168 |
Transformer | Dropout | dropout_4 | (None, 30, 30, 256) | 0 |
Decoder | Conv2DTranspose | conv2d_transpose | (None, 60, 60, 256) | 262400 |
Decoder | Concatenate | concatenate | (None, 60, 60, 512) | 0 |
Decoder | Conv2D | conv2d_6 | (None, 60, 60, 256) | 1179904 |
Decoder | Conv2D | conv2d_7 | (None, 60, 60, 256) | 590080 |
Decoder | Conv2DTranspose | conv2d_transpose_1 | (None, 120, 120, 128) | 131200 |
Decoder | Concatenate | concatenate_1 | (None, 120, 120, 256) | 0 |
Decoder | Conv2D | conv2d_8 | (None, 120, 120, 128) | 295040 |
Decoder | Conv2D | conv2d_9 | (None, 120, 120, 128) | 147584 |
Decoder | Conv2DTranspose | conv2d_transpose_2 | (None, 240, 240, 64) | 32832 |
Decoder | Concatenate | concatenate_2 | (None, 240, 240, 128) | 0 |
Decoder | Conv2D | conv2d_10 | (None, 240, 240, 64) | 73792 |
Decoder | Conv2D | conv2d 11 | (None, 240, 240, 64) | 36928 |
Output | Conv2D | conv2d_12 | (None, 240, 240, 4) | 260 |
Output | Activation | activation | (None, 240, 240, 4) | 0 |