--- a +++ b/scripts/test_segmentation.sh @@ -0,0 +1,19 @@ +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} \ No newline at end of file