-
Notifications
You must be signed in to change notification settings - Fork 48
/
Copy pathyolox-tiny.yaml
47 lines (43 loc) · 1.39 KB
/
yolox-tiny.yaml
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
__BASE__: [
'../coco.yaml',
'./hyp.scratch.yaml',
'./yolox-s.yaml'
]
# difference with s
# 1. no mixup
# 2. mosaic scale is 0.5
# 3. img_size is 416
# 4. multiscale range is (10, 20)
# 5. depth and width
img_size: 416
loss:
input_size: [ 416, 416 ]
network:
depth_multiple: 0.33
width_multiple: 0.375
data:
train_transforms: {
stage_epochs: [ 285, 15 ],
trans_list: [
[
{ func_name: mosaic, prob: 1.0 },
{ func_name: resample_segments },
{ func_name: random_perspective, prob: 1.0, degrees: 10.0, translate: 0.1, scale: 0.5, shear: 2.0 },
{ func_name: hsv_augment, prob: 1.0, hgain: 0.015, sgain: 0.7, vgain: 0.4 },
{ func_name: fliplr, prob: 0.5 },
{ func_name: label_norm, xyxy2xywh_: True },
{ func_name: label_pad, padding_size: 160, padding_value: -1 },
{ func_name: image_norm, scale: 255. },
{ func_name: image_transpose, bgr2rgb: True, hwc2chw: True },
],
[
{ func_name: letterbox, scaleup: False },
{ func_name: hsv_augment, prob: 1.0, hgain: 0.015, sgain: 0.7, vgain: 0.4 },
{ func_name: label_norm, xyxy2xywh_: True },
{ func_name: fliplr, prob: 0.5 },
{ func_name: label_pad, padding_size: 160, padding_value: -1 },
{ func_name: image_norm, scale: 255. },
{ func_name: image_transpose, bgr2rgb: True, hwc2chw: True },
]
]
}