Table 1.
Parameters of the Mobile U-Net model.
| Layer Type | Size | Filter | Stride |
|---|---|---|---|
| Input | 256 × 256 × 3/4 | ||
| Depthwise separable convolution | 256 × 256 × 64 | 3 × 3, 1 × 1 | 2 |
| Max pooling | 128 × 128 × 64 | 2 × 2 | 1 |
| Depthwise separable convolution | 128 × 128 × 128 | 3 × 3, 1 × 1 | 2 |
| Max pooling | 64 × 64 × 128 | 2 × 2 | 1 |
| Depthwise separable convolution | 64 × 64 × 256 | 3 × 3, 1 × 1 | 2 |
| Max pooling | 32 × 32 × 256 | 2 × 2 | 1 |
| Depthwise separable convolution | 32 × 32 × 512 | 3 × 3, 1 × 1 | 2 |
| Max pooling | 16 × 16 × 512 | 2 × 2 | 1 |
| Depthwise separable convolution | 16 × 16 × 1024 | 3 × 3, 1 × 1 | 2 |
| Transposed Convolution | 32 × 32 × 512 | 3 × 3 | 1 |
| Skip connection | 32 × 32 × 1024 | 1 | |
| Depthwise separable convolution | 32 × 32 × 512 | 3 × 3, 1 × 1 | 2 |
| Transposed Convolution | 64 × 64 × 256 | 3 × 3 | 1 |
| Skip connection | 64 × 64 × 512 | 1 | |
| Depthwise separable convolution | 64 × 64 × 256 | 3 × 3, 1 × 1 | 2 |
| Transposed Convolution | 128 × 128 × 128 | 3 × 3 | 1 |
| Skip connection | 128 × 128 × 256 | 1 | |
| Depthwise separable convolution | 128 × 128 × 128 | 3 × 3, 1 × 1 | 2 |
| Transposed Convolution | 256 × 256 × 64 | 3 × 3 | 1 |
| Skip connection | 256 × 256 × 128 | 1 | |
| Depthwise separable convolution | 256 × 256 × 64 | 3 × 3, 1 × 1 | 2 |
| Standard convolution | 256 × 256 × 3 | 1 × 1 | 1 |