-
Notifications
You must be signed in to change notification settings - Fork 99
/
Ocean.yaml
99 lines (91 loc) · 2.27 KB
/
Ocean.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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
OCEAN:
GPUS: '0,1,2,3,4,5,6,7'
PRINT_FREQ: 10
WORKERS: 32
OUTPUT_DIR: 'logs' # log file
CHECKPOINT_DIR: 'snapshot' # checkpoint file
TRAIN:
ISTRUE: True # whether to test
MODEL: "Ocean"
ALIGN: True # object aware branch
START_EPOCH: 0
END_EPOCH: 50
TEMPLATE_SIZE: 127
SEARCH_SIZE: 255
BATCH: 32
STRIDE: 8
RESUME: False
PRETRAIN: 'pretrain.model'
LR_POLICY: 'log'
WARMUP:
IFNOT: True
TYPE: 'step'
EPOCH: 5 # res50 5
KWARGS:
start_lr: 0.001 # res50 0.001 alex: 0.005
end_lr: 0.005 # res50 0.005 alex: 0.01
step: 1
LR:
TYPE: 'log'
KWARGS:
start_lr: 0.005 # res50 0.005 alex 0.01
end_lr: 0.00001 # res50 0.0005
LAYERS_LR: 0.1 # res50 0.1 alex: 1 # scale ration for backbone
BASE_LR: 0.005
UNFIX_EPOCH: 10
WARM_POLICY: 'step'
UNFIX_POLICY: 'log'
MOMENTUM: 0.9
WEIGHT_DECAY: 0.0001
TRAINABLE_LAYER: ['layer1', 'layer2', 'layer3'] # ['layer2', 'layer3', 'layer4']
WHICH_USE: ['YTB', 'VID', 'COCO', 'DET', 'GOT10K']
TEST: # TEST model is same as TRAIN.MODEL
ISTRUE: False # whether to test
THREADS: 16 # multi threads test
DATA: 'VOT2019'
START_EPOCH: 30
END_EPOCH: 50
RGBTSPLIT: None # None for main channel, 'RGB' and 'T' for RGBT
TUNE: # TUNE model is same as TRAIN.MODEL
ISTRUE: False # whether to tune
DATA: 'VOT2019'
METHOD: 'TPE'
RGBTSPLT: None
DATASET:
SHIFT: 4
SCALE: 0.05
COLOR: 1
FLIP: 0
BLUR: 0
ROTATION: 0
LABELSMOOTH: False
MIXUP: 0
GRAY: 0
CUTOUT: 0
SHIFTs: 64
SCALEs: 0.18
VID:
PATH: './data/vid/crop511'
ANNOTATION: './data/vid/train.json'
RANGE: 100
USE: 110000
YTB:
PATH: './data/y2b/crop511'
ANNOTATION: './data/y2b/train.json'
RANGE: 3
USE: 210000
GOT10K:
PATH: './data/got10k/crop511'
ANNOTATION: './data/got10k/all.json'
RANGE: 100
USE: 160000
DET:
PATH: './data/det/crop511'
ANNOTATION: './data/det/train.json'
RANGE: 100
USE: 60000
COCO:
PATH: "./data/coco/crop511"
ANNOTATION: "./data/coco/train2017.json"
RANGE: 1
USE: 60000