Table 4.
Parameter Name | Before Optimization | After Optimization |
---|---|---|
initial learning rate | 0.01 | 0.00816 |
final OneCycleLR learning | 0.2 | 0.25725 |
SGD momentum | 0.937 | 0.98 |
warmup_bias_lr | 0.1 | 0.11521 |
image HSV-Hue augmentation | 0.015 | 0.01734 |
image HSV-Saturation augmentation | 0.7 | 0.9 |
image HSV-Value augmentation | 0.4 | 0.44819 |
box loss gain | 0.05 | 0.03384 |
cls loss gain | 0.5 | 0.6195 |
warmup_epochs | 3.0 | 2.71044 |
warmup_momentum | 0.8 | 0.66111 |