-
Notifications
You must be signed in to change notification settings - Fork 23
/
config.py
63 lines (44 loc) · 946 Bytes
/
config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
anchors = [[1.3221, 1.73145], [3.19275, 4.00944], [5.05587, 8.09892], [9.47112, 4.84053], [11.2364, 10.0071]]
object_scale = 5
noobject_scale = 1
class_scale = 1
coord_scale = 1
saturation = 1.5
exposure = 1.5
hue = .1
jitter = 0.3
thresh = .6
batch_size = 16
lr = 0.0001
decay_lrs = {
60: 0.00001,
90: 0.000001
}
momentum = 0.9
weight_decay = 0.0005
# multi-scale training:
# {k: epoch, v: scale range}
multi_scale = True
# number of steps to change input size
scale_step = 40
scale_range = (3, 4)
epoch_scale = {
1: (3, 4),
15: (2, 5),
30: (1, 6),
60: (0, 7),
75: (0, 9)
}
input_sizes = [(320, 320),
(352, 352),
(384, 384),
(416, 416),
(448, 448),
(480, 480),
(512, 512),
(544, 544),
(576, 576)]
input_size = (416, 416)
test_input_size = (416, 416)
strides = 32
debug = False