Table 1. Details of our modified LeNet-5 convolutional neural network.
| Layer | Parameter | Output Shape | Numbera |
|---|---|---|---|
| Input | – | (None, 900, 2) | 0 |
| Conv1 | 32 × 5 × 2, stride 2, pad 0 | (None, 448, 32) | 352 |
| Max pooling2 | 3, stride 3, pad 0 | (None, 149, 32) | 0 |
| Conv3 | 64 × 5 × 2, stride 2, pad 0 | (None, 73, 64) | 10304 |
| Max pooling4 | 3, stride 3, pad 0 | (None, 24, 64) | 0 |
| Dropout5 | 0.8 rate | (None, 24, 64) | 0 |
| FC6 | 32, relu | (None, 32) | 49184 |
| Output | 2, softmax | (None, 2) | 66 |
Notes.
a
The number of parameters generated by the corresponding operation.