-
Notifications
You must be signed in to change notification settings - Fork 0
/
train_eval.sh
57 lines (48 loc) · 1.45 KB
/
train_eval.sh
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
exp="default"
gpu_num="4"
model="aott"
# model="aots"
# model="aotb"
# model="aotl"
# model="r50_aotl"
# model="swinb_aotl"
## Training ##
stage="pre"
python tools/train.py --amp \
--exp_name ${exp} \
--stage ${stage} \
--model ${model} \
--gpu_num ${gpu_num}
stage="pre_uvo_ovis"
python tools/train.py --amp \
--exp_name ${exp} \
--stage ${stage} \
--model ${model} \
--gpu_num ${gpu_num}
stage="pre_ytb_dav"
python tools/train.py --amp \
--exp_name ${exp} \
--stage ${stage} \
--model ${model} \
--gpu_num ${gpu_num}
## Evaluation ##
# dataset="davis2017"
# split="test"
# python tools/eval.py --exp_name ${exp} --stage ${stage} --model ${model} \
# --dataset ${dataset} --split ${split} --gpu_num ${gpu_num}
# dataset="davis2017"
# split="val"
# python tools/eval.py --exp_name ${exp} --stage ${stage} --model ${model} \
# --dataset ${dataset} --split ${split} --gpu_num ${gpu_num}
# dataset="davis2016"
# split="val"
# python tools/eval.py --exp_name ${exp} --stage ${stage} --model ${model} \
# --dataset ${dataset} --split ${split} --gpu_num ${gpu_num}
# dataset="youtubevos2018"
# split="val" # or "val_all_frames"
# python tools/eval.py --exp_name ${exp} --stage ${stage} --model ${model} \
# --dataset ${dataset} --split ${split} --gpu_num ${gpu_num}
dataset="youtubevos2019"
split="val" # or "val_all_frames"
python tools/eval.py --exp_name ${exp} --stage ${stage} --model ${model} \
--dataset ${dataset} --split ${split} --gpu_num ${gpu_num}