Table 2.
The number of channels and output dimensions of each layer.
| Layer name | Operate | Kernel size | Stride | Output size | Channels |
|---|---|---|---|---|---|
| Input | 224 × 224 | 3 | |||
| Top layer | conv2d | 7 × 7 | 2 | 112 × 112 | 16 |
| Max Pool2d | 3 × 3 | 2 | 56 × 56 | 16 | |
| R-block1 | 1 1 1 |
56 × 56 | 16 | ||
| Middle layer | R-block5 | 1 1 1 |
56 × 56 | 16 | |
| Bottom layer | RA-UNET | 56 × 56 | 16 | ||
| Top layer | R-block2 | 1 2 1 |
28 × 28 | 32 | |
| R-block3 | 1 2 1 |
14 × 14 | 64 | ||
| R-block4 | 1 2 1 |
7 × 7 | 64 | ||
| Avg Pool2d | 7 × 7 | 1 | 1 × 1 | 64 | |
| Output | 4 |