Table 2 Network training hyperparameters.
From: Detection of safety helmet and mask wearing using improved YOLOv5s
Name of the training parameter | Parameter value |
|---|---|
Initial learning rate | 0.01 |
Weight attenuation factor | 0.0005 |
Momentum | 0.937 |
Batch | 16 |
Epoch | 300 |
