IBM_AI/vi-opton/opton_cls_train/eval.sh

set -e

###########################################################
#
# ./eval.sh IMG_DIR SNAPSHOT_ITER [TASK_NAME] [cpu]
#
###########################################################

#export GLOG_minloglevel=1

CAFFE_HOME=/mnt/sdb/tools/caffe-bvlc

export LD_LIBRARY_PATH=$CAFFE_HOME/build/lib:$LD_LIBRARY_PATH
export PYTHONPATH=$CAFFE_HOME/python:$PYTHONPATH

# NET='bvlc_alexnet'
NET='bvlc_googlenet'

ROOT_DIR=$(cd `dirname $0`; pwd)

LOG_PATH=$ROOT_DIR/log/${NET}
mkdir -p $LOG_PATH
LOG="$LOG_PATH/eval_`date +'%Y-%m-%d_%H-%M-%S'`.txt"
exec &> >(tee -a "$LOG")
echo Logging output to "$LOG"

IMG_DIR=$1

SNAPSHOT_ITER=100000
if [ $# -gt 1 ]; then
    SNAPSHOT_ITER=$2
fi

TASK_NAME=
if [ $# -gt 2 ]; then
    TASK_NAME=$3
fi

# 1,use CPU; 0,use GPU;
FORCE_CPU=0         
if [ $# -gt 3 ]; then
    if [ $4 == "cpu" ];then
        FORCE_CPU=1
    fi
fi

CAFFEMODEL=$ROOT_DIR/output/snapshot/${NET}/${NET}_iter_${SNAPSHOT_ITER}.caffemodel
PROTOTXT=$ROOT_DIR/prototxt/${NET}/deploy.prototxt

time python $ROOT_DIR/eval.py \
--force_cpu ${FORCE_CPU} \
--def ${PROTOTXT} \
--net ${CAFFEMODEL} \
--mean $ROOT_DIR/data/images_mean.npy \
--pred_dir $ROOT_DIR/output/eval/${NET}/${SNAPSHOT_ITER}/${TASK_NAME} \
--img_dir ${IMG_DIR} \
--save_img True \
--save_mark True \
# 2>&1|tee $LOG