pull/2091/head
HydrogenSulfate 2022-06-21 12:41:54 +00:00
parent 223872ae10
commit f55172394b
6 changed files with 10 additions and 10 deletions

View File

@ -208,7 +208,7 @@ fi
if [[ ${MODE} = "serving_infer" ]]; then if [[ ${MODE} = "serving_infer" ]]; then
# prepare serving env # prepare serving env
python_name=$(func_parser_value "${lines[2]}") python_name=$(func_parser_value "${lines[2]}")
if [[ ${model_name} =~ "ShiTu" ]]; then if [[ ${model_name} = "PPShiTu" ]]; then
cls_inference_model_url=$(func_parser_value "${lines[3]}") cls_inference_model_url=$(func_parser_value "${lines[3]}")
cls_tar_name=$(func_get_url_file_name "${cls_inference_model_url}") cls_tar_name=$(func_get_url_file_name "${cls_inference_model_url}")
det_inference_model_url=$(func_parser_value "${lines[4]}") det_inference_model_url=$(func_parser_value "${lines[4]}")

View File

@ -237,14 +237,14 @@ echo "################### build PaddleClas demo finished ###################"
# set cuda device # set cuda device
# GPUID=$2 GPUID=$3
# if [ ${#GPUID} -le 0 ];then if [ ${#GPUID} -le 0 ];then
# env="export CUDA_VISIBLE_DEVICES=0" env="export CUDA_VISIBLE_DEVICES=0"
# else else
# env="export CUDA_VISIBLE_DEVICES=${GPUID}" env="export CUDA_VISIBLE_DEVICES=${GPUID}"
# fi fi
# set CUDA_VISIBLE_DEVICES set CUDA_VISIBLE_DEVICES
# eval $env eval $env
echo "################### run test ###################" echo "################### run test ###################"

View File

@ -310,7 +310,7 @@ echo "################### run test ###################"
export Count=0 export Count=0
IFS="|" IFS="|"
if [[ ${model_name} =~ "ShiTu" ]]; then if [[ ${model_name} = "PPShiTu" ]]; then
func_serving_rec func_serving_rec
else else
func_serving_cls func_serving_cls