提交 38e78b0c 编写于 作者: H HydrogenSulfate

add paddle2onnx_infer chain

上级 a8278e84
===========================paddle2onnx_params===========================
model_name:GeneralRecognitionV2_PPLCNetV2_base
python:python3.7
2onnx: paddle2onnx
--model_dir:./deploy/models/general_PPLCNetV2_base_pretrained_v1.0_infer/
--model_filename:inference.pdmodel
--params_filename:inference.pdiparams
--save_file:./deploy/models/general_PPLCNetV2_base_pretrained_v1.0_infer/inference.onnx
--opset_version:10
--enable_onnx_checker:True
inference_model_url:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/inference/PP-ShiTuV2/general_PPLCNetV2_base_pretrained_v1.0_infer.tar
inference:./python/predict_rec.py
Global.use_onnx:True
Global.rec_inference_model_dir:./models/general_PPLCNetV2_base_pretrained_v1.0_infer
Global.use_gpu:False
-c:configs/inference_rec.yaml
\ No newline at end of file
...@@ -321,6 +321,7 @@ if [[ ${MODE} = "paddle2onnx_infer" ]]; then ...@@ -321,6 +321,7 @@ if [[ ${MODE} = "paddle2onnx_infer" ]]; then
inference_model_url=$(func_parser_value "${lines[10]}") inference_model_url=$(func_parser_value "${lines[10]}")
tar_name=${inference_model_url##*/} tar_name=${inference_model_url##*/}
${python_name} -m pip install onnx
${python_name} -m pip install paddle2onnx ${python_name} -m pip install paddle2onnx
${python_name} -m pip install onnxruntime ${python_name} -m pip install onnxruntime
if [[ ${model_name} =~ "GeneralRecognition" ]]; then if [[ ${model_name} =~ "GeneralRecognition" ]]; then
...@@ -337,14 +338,12 @@ if [[ ${MODE} = "paddle2onnx_infer" ]]; then ...@@ -337,14 +338,12 @@ if [[ ${MODE} = "paddle2onnx_infer" ]]; then
rm -rf val_list.txt rm -rf val_list.txt
ln -s demo_test.txt val_list.txt ln -s demo_test.txt val_list.txt
cd ../../ cd ../../
eval "wget -nc $model_url_value --no-check-certificate"
mv general_PPLCNet_x2_5_pretrained_v1.0.pdparams GeneralRecognition_PPLCNet_x2_5_pretrained.pdparams
fi fi
cd deploy cd deploy
mkdir models mkdir models
cd models cd models
wget -nc ${inference_model_url} wget -nc ${inference_model_url}
tar xf ${tar_name} eval "tar -xf ${tar_name}"
cd ../../ cd ../../
fi fi
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册