From f7a554c2af9cff0a40ea70abb01ca77c0884e1c9 Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Tue, 8 Jun 2021 20:42:11 +0800 Subject: [PATCH] fix det_model_dir rec_model_dir Null --- test/test.sh | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/test/test.sh b/test/test.sh index cd4d93d9..b43b1a61 100644 --- a/test/test.sh +++ b/test/test.sh @@ -116,8 +116,12 @@ for train_model in ${train_model_list[*]}; do if [ "${model_name}" = "det" ]; then export rec_batch_size_list=( "1" ) inference="tools/infer/predict_det.py" + det_model_dir=${save_log}/export_inference/ + rec_model_dir="" elif [ "${model_name}" = "rec" ]; then inference="tools/infer/predict_rec.py" + rec_model_dir=${save_log}/export_inference/ + det_model_dir="" fi # inference for device in ${devices[*]}; do