Research Article
Railway Fastener Fault Diagnosis Based on Generative Adversarial Network and Residual Network Model
Table 3
Discriminator network algorithm.
| | Discriminator network algorithm |
| | Input: Fractured track fastener picture | | Output: Probability that the output data is true or false | | 1 Initialization of relevant parameters | | Set up the training cycle (e): epoch | | Set up the batch sample size (m): batch_size | | 2 Begin | | 3 For i = 1 in e do | | 4 For batch = 1 in m do | | 5 | | 6 Optimize loss functions | | 7 Update weight and bias | | 8 End | | 9 End | | 10 End |
|
|