Table 5 Layer structure of the LeNet-5 CNN employed in this investigation
Layers | Type | Parameters |
|---|---|---|
Image | ||
1 | Input | 3-tensor (size: 1 × 28 × 28) |
2 | Convolution | 3-tensor (size: 10 × 25 × 25) |
3 | Ramp | 3-tensor (size: 10 × 25 × 25) |
4 | Pooling | 3-tensor (size: 10 × 12 × 12) |
5 | Convolution | 3-tensor (size: 20 × 9 × 9) |
6 | Ramp | 3-tensor (size: 20 × 9 × 9) |
7 | Pooling | 3-tensor (size: 20 × 4 × 4) |
8 | Flatten | Vector (size: 320) |
9 | Linear | Vector (size: 2) |
10 | Output | Vector (size: 2) |