-
Notifications
You must be signed in to change notification settings - Fork 15
/
do_test.sh
executable file
·32 lines (24 loc) · 1.01 KB
/
do_test.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
rootpath=$HOME/VisualSearch
overwrite=0
#trainCollection=tgif-msrvtt10k
#valCollection=tv2016train
#val_set=setA
#config=w2vvpp_resnext101-resnet152_subspace
#model_path=$rootpath/$trainCollection/w2vvpp_train/$valCollection/$val_set/$config/runs_0/model_best.pth.tar
#sim_name=$trainCollection/$valCollection/$val_set/$config
#./do_test.sh iacc.3 ~/VisualSearch/tgif-msrvtt10k/w2vvpp_train/tv2016train/setA/w2vvpp_resnext101-resnet152_subspace/runs_0/model_best.pth.tar tgif-msrvtt10k/tv2016train/setA/w2vvpp_resnext101-resnet152_subspace/runs_0
if [ "$#" -ne 4 ]; then
echo "Usage: $0 testCollection model_path sim_name query_sets"
exit
fi
testCollection=$1
model_path=$2
sim_name=$3
query_sets=$4 # tv16.avs.txt,tv17.avs.txt,tv18.avs.txt for TRECVID 16/17/18 and tv19.avs.txt for TRECVID19
if [ ! -f "$model_path" ]; then
echo "model not found: $model_path"
exit
fi
python predictor.py $testCollection $model_path $sim_name \
--query_sets $query_sets \
--rootpath $rootpath --overwrite $overwrite