Layer | Output dimension (image data) | Output dimension (IQ data) |
---|---|---|
Input | 2×128×128 | 1×2×128 |
Conv1 | 64×64×64 | 64×1×64 |
Conv2 | 64×32×32 | 64×1×32 |
Conv3 | 128×16×16 | 128×1×16 |
Conv4 | 256×8×8 | 256×1×8 |
Conv5 | 512×4×4 | 512×1×4 |
Average pool | 512×1×1 | 512×1×1 |
Fully connected | 4 | 4 |
Softmax | 4 | 4 |