超参数

数值

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