Table A3.
Architecture of our U-Net. #Params: Number of parameters.
| Layer | Output Shape | # Params |
|---|---|---|
| Input | (256, 256, 1) | 0 |
| Convolution (strides = (1, 1), filters = 16, kernel = (3, 3), activation = ELU) | (256, 256, 16) | 160 |
| Dropout (0.1) | (256, 256, 16) | 0 |
| Convolution (strides = (1, 1), filters = 16, kernel = (3, 3), activation = ELU, name = C1) | (256, 256, 16) | 2320 |
| Convolution (strides = (2, 2), filters = 16, kernel = (3, 3), activation = ELU) | (128, 128, 16) | 2320 |
| Convolution (strides = (1, 1), filters = 32, kernel = (3, 3), activation = ELU) | (128, 128, 32) | 4640 |
| Dropout (0.1) | (128, 128, 32) | 0 |
| Convolution (strides = (1, 1), filters = 32, kernel = (3, 3), activation = ELU, name = C2) | (128, 128, 32) | 9248 |
| Convolution (strides = (2, 2), filters = 32, kernel = (3, 3), activation = ELU) | (64, 64, 32) | 9248 |
| Convolution (strides = (1, 1), filters = 64, kernel = (3, 3), activation = ELU) | (64, 64, 64) | 18,496 |
| Dropout (0.2) | (64, 64, 64) | 0 |
| Convolution (strides = (1, 1), filters = 64, kernel = (3, 3), activation = ELU, name = C3) | (64, 64, 64) | 36,928 |
| Convolution (strides = (2, 2), filters = 64, kernel = (3, 3), activation = ELU) | (32, 32, 64) | 36,928 |
| Convolution (strides = (1, 1), filters = 128, kernel = (3, 3), activation = ELU) | (32, 32, 128) | 73,856 |
| Dropout (0.2) | (32, 32, 128) | 0 |
| Convolution (strides = (1, 1), filters = 128, kernel = (3, 3), activation = ELU, name = C4) | (32, 32, 128) | 147,584 |
| Convolution (strides = (2, 2), filters = 128, kernel = (3, 3), activation = ELU) | (16, 16, 128) | 147,584 |
| Convolution (strides = (1, 1), filters = 256, kernel = (3, 3), activation = ELU) | (16, 16, 256) | 295,168 |
| Dropout (0.3) | (16, 16, 256) | 0 |
| Convolution (strides = (1, 1), filters = 256, kernel = (3, 3), activation = ELU, name = C5) | (16, 16, 256) | 590,080 |
| Convolution (strides = (2, 2), filters = 256, kernel = (3, 3), activation = ELU) | (8, 8, 256) | 590,080 |
| Convolution (strides = (1, 1), filters = 256, kernel = (3, 3), activation = ELU) | (8, 8, 256) | 590,080 |
| Dropout (0.3) | (8, 8, 256) | 0 |
| Convolution (strides = (1, 1), filters = 256, kernel = (3, 3), activation = ELU) | (8, 8, 256) | 590,080 |
| Transposed Convolution (strides = (2, 2), filters = 256, kernel = (2, 2), activation = ELU, name = U1) | (16, 16, 256) | 262,400 |
| Concatenation (C5, U1) | (16, 16, 512) | 0 |
| Convolution (strides = (1, 1), filters = 256, kernel = (3, 3), activation = ELU) | (16, 16, 256) | 1,179,904 |
| Dropout (0.3) | (16, 16, 256) | 0 |
| Convolution (strides = (1, 1), filters = 256, kernel = (3, 3), activation = ELU) | (16, 16, 256) | 590,080 |
| Transposed Convolution (strides = (2, 2), filters = 128, kernel = (2, 2), activation = ELU, name = U2) | (32, 32, 128) | 131,200 |
| Concatenation (C4, U2) | (32, 32, 256) | 0 |
| Convolution (strides = (1, 1), filters = 128, kernel = (3, 3), activation = ELU) | (32, 32, 128) | 295,040 |
| Dropout (0.2) | (32, 32, 128) | 0 |
| Convolution (strides = (1, 1), filters = 128, kernel = (3, 3), activation = ELU) | (32, 32, 128) | 147,584 |
| Transposed Convolution (strides = (2, 2), filters = 64, kernel = (2, 2), activation = ELU, name = U3) | (64, 64, 64) | 32,832 |
| Concatenation (C3, U3) | (64, 64, 128) | 0 |
| Convolution (strides = (1, 1), filters = 64, kernel = (3, 3), activation = ELU) | (64, 64, 64) | 73,792 |
| Dropout (0.2) | (64, 64, 64) | 0 |
| Convolution (strides = (1, 1), filters = 64, kernel = (3, 3), activation = ELU) | (64, 64, 64) | 36,928 |
| Transposed Convolution (strides = (2, 2), filters = 32, kernel = (2, 2), activation = ELU, name = U4) | (128, 128, 32) | 8224 |
| Concatenation (C2, U4) | (128, 128, 64) | 0 |
| Convolution (strides = (1, 1), filters = 32, kernel = (3, 3), activation = ELU) | (128, 128, 32) | 18,464 |
| Dropout (0.1) | (128, 128, 32) | 0 |
| Convolution (strides = (1, 1), filters = 32, kernel = (3, 3), activation = ELU) | (128, 128, 32) | 9248 |
| Transposed Convolution (strides = (2, 2), filters = 16, kernel = (2, 2), activation = ELU, name = U5) | (256, 256, 16) | 2064 |
| Concatenation (C1, U5) | (256, 256, 16) | 0 |
| Convolution (strides = (1, 1), filters = 16, kernel = (3, 3), activation = ELU) | (256, 256, 16) | 4624 |
| Dropout (0.1) | (256, 256, 16) | 0 |
| Convolution (strides = (1, 1), filters = 16, kernel = (3, 3), activation = ELU) | (256, 256, 16) | 2320 |
| Convolution (strides = (1, 1), filters = 1, kernel = (1, 1), activation = Sigmoid, name = Output) | (256, 256, 1) | 17 |