Research Article
Defect Diagnosis of Gear-Shaft Bearing System Based on the OWF-TSCNN Composed of Wavelet Time-Frequency Map and FFT Spectrum 1
Table 2
2D-CNN network parameter settings.
| Serial number | Layer | Convolution kernels/step | Convolution kernels | Output size | Activation function |
| 1 | Input layer | | | | | 2 | Conv-2D-1 | | 6 | | ReLU | 3 | Pooling-2D-1 | | 6 | | | 4 | BN-2D-1 | | | | | 5 | Conv-2D-2 | | 16 | | ReLU | 6 | Pooling-2D-2 | | 16 | | | 7 | BN-2D-2 | | | | | 8 | Flatten-2D | | | | |
|
|