Table 4 The number of learnable parameters in the each layer of a U-Net model.
From: 3D brain tumor segmentation using a two-stage optimal mass transport algorithm
Layer | Dimension | Learnable parameters |
|---|---|---|
Encoder-1 | 128 | 59,040 |
Encoder-2 | 64 | 332,352 |
Encoder-3 | 32 | 1,328,256 |
Bridge-1 | 16 | 1,770,240 |
Bridge-2 | 16 | 3,540,480 |
Decoder-1 | 32 | 9,177,088 |
Decoder-2 | 64 | 2,294,784 |
Decoder-3 | 128 | 573,952 |
FC | 128 | 130 |
