Skip to content

GitLab

Commit f7a554c2 authored by LDOUBLEV's avatar LDOUBLEV
Browse files

fix det_model_dir rec_model_dir Null

parent 81f2ccb1
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
test/test.sh
View file @ f7a554c2
...@@ -116,8 +116,12 @@ for train_model in ${train_model_list[*]}; do ...@@ -116,8 +116,12 @@ for train_model in ${train_model_list[*]}; do
if [ "${model_name}" = "det" ]; then if [ "${model_name}" = "det" ]; then
export rec_batch_size_list=( "1" ) export rec_batch_size_list=( "1" )
inference="tools/infer/predict_det.py" inference="tools/infer/predict_det.py"
det_model_dir=${save_log}/export_inference/
rec_model_dir=""
elif [ "${model_name}" = "rec" ]; then elif [ "${model_name}" = "rec" ]; then
inference="tools/infer/predict_rec.py" inference="tools/infer/predict_rec.py"
rec_model_dir=${save_log}/export_inference/
det_model_dir=""
fi fi
# inference # inference
for device in ${devices[*]}; do for device in ${devices[*]}; do
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment