Table 1.
The parameters of each level for CNN.
Layer Type | Output Shape | Param | |
---|---|---|---|
Input Layer | Input | 64 × 64 × 3 | 0 |
Hidden Layer 1 | Conv1 | 64 × 64 × 32 | 896 |
ReLU | 64 × 64 × 32 | 0 | |
Pool1 | 32 × 32 × 32 | 0 | |
Hidden Layer 2 | Conv2 | 32 × 32 × 64 | 18,496 |
ReLU | 32 × 32 × 64 | 0 | |
Pool2 | 16 × 16 × 64 | 0 | |
Hidden Layer 3 | Conv3 | 16 × 16 × 128 | 73,856 |
ReLU | 16 × 16 × 128 | 0 | |
Pool3 | 8 × 8 × 128 | 0 | |
Classification layer | Flatten | 8192 | 0 |
Dense1 | 16 | 131,088 | |
ReLU | 16 | 0 | |
Dense2 | 64 | 1088 | |
ReLU | 64 | 0 | |
Dense3 | 128 | 8320 | |
ReLU | 128 | 0 | |
Dense4 | 2 | 258 |