set -ex

DATAROOT=$1

DATAFORMAT=$2

RESULTS_DIR=$3

#CARSON_PATH='./pretrained_models/carson_Jan2021.h5'

CARSON_PATH='../private_models/main_carson_model.h5'

CARMEN_PATH='./pretrained_models/carmen_Jan2021.h5'

PIPELINE='segmentation'

# command

CUDA_VISIBLE_DEVICES=${GPU_ID} python ./test.py \

  --dataroot ${DATAROOT} \

  --dataformat ${DATAFORMAT} \

  --results_dir ${RESULTS_DIR} \

  --pretrained_models_netS ${CARSON_PATH} \

  --pretrained_models_netME ${CARMEN_PATH} \

  --pipeline ${PIPELINE}