提交 015a8f06 编写于 作者: L lijianshe02

support variable length input

上级 a4138a62
......@@ -37,21 +37,5 @@ if [ "$mode"x == "predict"x ]; then
--use_gpu=$use_gpu \
--video_path=''
fi
elif [ "$mode"x == "inference"x ]; then
echo $mode $name $configs $weights
if [ "$weights"x != ""x ]; then
python inference_model.py --model_name=$name \
--config=$configs \
--weights=$weights \
--use_gpu=$use_gpu \
--save_dir=$save_inference_dir
else
python inference_model.py --model_name=$name \
--config=$configs \
--use_gpu=$use_gpu \
--save_dir=$save_inference_dir
fi
else
echo "Not implemented mode " $mode
fi
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册