Table 3.
Layer | Input Dimensions | Output Channels |
---|---|---|
Conv2d | 224 × 224 × 3 | 32 |
Bn | 112 × 112 × 32 | 16 |
Bn | 112 × 112 × 16 | 24 |
Bn | 56 × 56 × 24 | 32 |
Bn | 28 × 28 × 32 | 64 |
Bn | 14 × 14 × 64 | 96 |
Bn | 14 × 14 × 96 | 160 |
Bn | 7 × 7 × 160 | 320 |
Conv2d | 7 × 7 × 320 | 1280 |
Avgpool | 7 × 7 × 1280 | 1280 |
Flatten | 1 × 1 × 1280 | - |
Dense 1 | 1280 | 256 |
Dense 2 | 256 | 128 |
Dense 3 | 128 | 2 |