Spaces:
Sleeping
Sleeping
File size: 1,898 Bytes
d4b77ac |
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 |
show_help() {
cat << EOF
Usage:
${0##*/} [-h/--help] [-s/--start] [-e/--end] [-d/--dataset] [-m/--model] [-g/--gpu]
e.g.
bash ${0##*/} -s 1 -e 20 -d VOT2018 -g 4 # for test models
bash ${0##*/} -m snapshot/checkpoint_e10.pth -n 8 -g 4 # for tune models
EOF
}
ROOT=`git rev-parse --show-toplevel`
source activate siammask
export PYTHONPATH=$ROOT:$PYTHONPATH
export PYTHONPATH=$PWD:$PYTHONPATH
dataset=VOT2018
NUM=4
START=1
END=20
GPU=0
while [[ $# -gt 0 ]]
do
key="$1"
case $key in
-h|--help)
show_help
exit
;;
-d|--dataset)
dataset=$2
shift 2
;;
-n|--num)
NUM=$2
shift 2
;;
-s|--start)
START=$2
shift 2
;;
-e|--end)
END=$2
shift 2
;;
-m|--model)
model=$2
shift 2
;;
-g|--gpu)
GPU=$2
shift 2
;;
*)
echo invalid arg [$1]
show_help
exit 1
;;
esac
done
set -e
if [ -z "$model" ]; then
echo test snapshot $START ~ $END on dataset $dataset with $GPU gpus.
for i in $(seq $START $END)
do
bash test.sh snapshot/checkpoint_e$i.pth $dataset $(($i % $GPU)) &
done
wait
python $ROOT/tools/eval.py --dataset $dataset --num 20 --tracker_prefix C --result_dir ./test/$dataset 2>&1 | tee logs/eval_test_$dataset.log
else
echo tuning $model on dataset $dataset with $NUM jobs in $GPU gpus.
for i in $(seq 1 $NUM)
do
bash tune.sh $model $dataset $(($i % $GPU)) &
done
wait
rm finish.flag
python $ROOT/tools/eval.py --dataset $dataset --num 20 --tracker_prefix C --result_dir ./result/$dataset 2>&1 | tee logs/eval_tune_$dataset.log
fi
|