Table 3.
Stage | Operator/Block | Resolution | Channels | Layers |
---|---|---|---|---|
1 | Conv 3 × 3 + BN + Swish | 224 × 224 | 48 | 1 |
2 | MBConv1, k3 × 3 | 224 × 224 | 24 | 3 |
3 | MBConv6, k5 × 5 | 112 × 112 | 40 | 5 |
4 | MBConv6, k5 × 5 | 56 × 56 | 64 | 5 |
5 | MBConv6, k5 × 5 | 28 × 28 | 128 | 7 |
6 | MBConv6, k3 × 3 | 28 × 28 | 176 | 7 |
7 | MBConv6, k5 × 5 | 14 × 14 | 304 | 9 |
8 | MBConv6, k3 × 3 | 14 × 14 | 512 | 3 |
9 | Conv 1 × 1 | 14 × 14 | 2048 | 1 |
10 | BN | 14 × 14 | 2048 | 1 |
11 | Activation | 14 × 14 | 2048 | 1 |
12 | Global average pooling | 2048 | 1 | 1 |
13 | Dropout | 2048 | 1 | 1 |
14 | Dense | 1 | 1 | 1 |