fix
parent
223872ae10
commit
f55172394b
|
@ -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]}")
|
||||||
|
|
|
@ -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 ###################"
|
||||||
|
|
|
@ -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
|
||||||
|
|
Loading…
Reference in New Issue