elia / scripts /test_elia.py
yxchng
add files
a166479
import os
import sys
refcoco_ckpt_path = "model_best_refcoco_0508.pth"
refcocop_ckpt_path = "model_best_refcoco+_0508.pth"
cmd1 = 'python test_elia.py --model="lavt" --swin_type="base" --dataset="refcoco" --split="val" --resume="{:s}" --workers="4" --ddp_trained_weights --window12 --img_size="480"'.format(refcoco_ckpt_path)
cmd2 = 'python test_elia.py --model="lavt" --swin_type="base" --dataset="refcoco" --split="testA" --resume="{:s}" --workers="4" --ddp_trained_weights --window12 --img_size="480"'.format(refcoco_ckpt_path)
cmd3 = 'python test_elia.py --model="lavt" --swin_type="base" --dataset="refcoco" --split="testB" --resume="{:s}" --workers="4" --ddp_trained_weights --window12 --img_size="480"'.format(refcoco_ckpt_path)
cmd4 = 'python test_elia.py --model="lavt" --swin_type="base" --dataset="refcoco+" --split="val" --resume="{:s}" --workers="4" --ddp_trained_weights --window12 --img_size="480"'.format(refcocop_ckpt_path)
cmd5 = 'python test_elia.py --model="lavt" --swin_type="base" --dataset="refcoco+" --split="testA" --resume="{:s}" --workers="4" --ddp_trained_weights --window12 --img_size="480"'.format(refcocop_ckpt_path)
cmd6 = 'python test_elia.py --model="lavt" --swin_type="base" --dataset="refcoco+" --split="testB" --resume="{:s}" --workers="4" --ddp_trained_weights --window12 --img_size="480"'.format(refcocop_ckpt_path)
cmds = [cmd1,cmd2,cmd3,cmd4,cmd5,cmd6]
for cmd in cmds:
print(cmd, flush=True)
os.system(cmd)