Skip to main content
. 2024 May 23;24(11):3321. doi: 10.3390/s24113321

Table 4.

Hyperparameter optimization results.

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