-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy pathingraph-fromscratch-eval-test-np-cider.sh
40 lines (31 loc) · 1.36 KB
/
ingraph-fromscratch-eval-test-np-cider.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
#!/bin/bash
DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
model_name="show_and_tell_in_graph_model_fromscratch"
num_processes=1
device=1
model=ShowAndTellInGraphModel
MODEL_DIR="${DIR}/model/${model_name}"
for ckpt in $(ls ${MODEL_DIR} | python ${DIR}/tools/every_n_step.py 20000 | tail -n 15 | tac); do
# the script directory
VALIDATE_IMAGE_DIR="${DIR}/data/ai_challenger_caption_validation_20170910/caption_validation_images_20170910"
VALIDATE_REFERENCE_FILE="${DIR}/data/ai_challenger_caption_validation_20170910/reference.json"
CHECKPOINT_PATH="${MODEL_DIR}/model.ckpt-$ckpt"
OUTPUT_DIR="${MODEL_DIR}/model.ckpt-${ckpt}.eval"
mkdir $OUTPUT_DIR
cd ${DIR}/im2txt
if [ ! -f ${OUTPUT_DIR}/out.json ]; then
CUDA_VISIBLE_DEVICES=$device python inference.py \
--input_file_pattern="${VALIDATE_IMAGE_DIR}/*.jpg" \
--checkpoint_path=${CHECKPOINT_PATH} \
--vocab_file=${DIR}/data/word_counts.txt \
--output=${OUTPUT_DIR}/out.json \
--model=${model} \
--support_ingraph=True
echo output saved to ${OUTPUT_DIR}/out.json
fi
if [ ! -f ${OUTPUT_DIR}/out.eval ]; then
python ${DIR}/tools/eval/run_evaluations.py --submit ${OUTPUT_DIR}/out.json --ref $VALIDATE_REFERENCE_FILE 2> ${OUTPUT_DIR}/err.eval | tee ${OUTPUT_DIR}/out.eval | grep ^Eval
echo eval result saved to ${OUTPUT_DIR}/out.eval
fi
exit 1
done