Skip to main content
. 2021 Jul 16;11:14629. doi: 10.1038/s41598-021-94190-9

Table 2.

Network structure of the U-Net architecture.

Module Layer name Kernel size Stride Kernel number Output size
Input 128 × 128 × 4
Conv 1 3 × 3 1 16 128 × 128 × 16
Conv 2 3 × 3 1 16 128 × 128 × 16
Max-pooling 2 × 2 2 16 64 × 64 × 16
Conv 3 3 × 3 1 32 64 × 64 × 32
Conv 4 3 × 3 1 32 64 × 64 × 32
Max-pooling 2 × 2 2 32 32 × 32 × 32
Conv 5 3 × 3 1 64 32 × 32 × 64
Conv 6 3 × 3 1 64 32 × 32 × 64
Encoding Max-pooling 2 × 2 2 64 16 × 16 × 64
Conv 7 3 × 3 1 128 16 × 16 × 128
Conv 8 3 × 3 1 128 16 × 16 × 128
Max-pooling 2 × 2 2 128 8 × 8 × 128
Conv 9 3 × 3 1 256 8 × 8 × 256
Conv 10 3 × 3 1 256 8 × 8 × 256
Conv-transpose 3 × 3 2 128 16 × 16 × 128
Conv 11 3 × 3 1 128 16 × 16 × 128
Decoding Conv 12 3 × 3 1 128 16 × 16 × 128
Conv-transpose 3 × 3 2 64 32 × 32 × 64
Conv 13 3 × 3 1 64 32 × 32 × 64
Conv 14 3 × 3 1 64 32 × 32 × 64
Conv-transpose 3 × 3 2 32 64 × 64 × 32
Conv 15 3 × 3 1 32 64 × 64 × 32
Conv 16 3 × 3 1 32 64 × 64 × 32
Conv-transpose 3 × 3 2 16 128 × 128 × 16
Conv 17 3 × 3 1 16 128 × 128 × 16
Conv 18 3 × 3 1 16 128 × 128 × 16
Output Conv 19 1 × 1 1 1 128 × 128 × 1