超参数 | 数值 |
Learning rate | 0.01 |
SGD momentum | 0.937 |
optimizer weight decay | 0.0005 |
ciou loss gain | 0.05 |
cls loss gain | 0.5 |
cls BCELoss positive weight | 1.0 |
obj loss gain | 1.0 |
obj BCELoss positive weight | 1.0 |
iou training threshold | 0.80 |
anchor-multiple threshold | 4.0 |
image HSV-Hue augmentation | 0.015 |
image HSV-Saturation augmentation | 0.7 |
image HSV-Value augmentation | 0.4 |
image scale | 0.5 |
batch | 16 |
epoch | 100 |