File size: 1,316 Bytes
c985ba4 |
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 |
exp="default"
gpu_num="4"
model="aott"
# model="aots"
# model="aotb"
# model="aotl"
# model="r50_deaotl"
# model="swinb_aotl"
## Training ##
stage="pre"
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} |