提交 8ffef11f 编写于 作者: A andyjpaddle

update for tipc

上级 b7656e68
......@@ -9,7 +9,7 @@ inference:./deploy/cpp_infer/build/ppocr --rec_char_dict_path=./ppocr/utils/ppoc
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16
--precision:fp32
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
--rec_model_dir:./inference/ch_PP-OCRv2_rec_infer/
......
......@@ -9,7 +9,7 @@ inference:./deploy/cpp_infer/build/ppocr
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16
--precision:fp32
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
null:null
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_det.py
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16|int8
--precision:fp32
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
null:null
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_det.py
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16|int8
--precision:fp32
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
null:null
......
......@@ -9,7 +9,7 @@ inference:./deploy/cpp_infer/build/ppocr --rec_char_dict_path=./ppocr/utils/ppoc
--cpu_threads:6
--rec_batch_num:6
--use_tensorrt:False
--precision:fp32|fp16
--precision:fp32
--rec_model_dir:
--image_dir:./inference/rec_inference/
null:null
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_rec.py
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:False
--precision:fp32|int8
--precision:fp32
--rec_model_dir:
--image_dir:./inference/rec_inference
null:null
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_rec.py
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:False
--precision:fp32|int8
--precision:fp32
--rec_model_dir:
--image_dir:./inference/rec_inference
null:null
......
......@@ -9,7 +9,7 @@ inference:./deploy/cpp_infer/build/ppocr --rec_img_h=48 --rec_char_dict_path=./p
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16
--precision:fp32
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
--rec_model_dir:./inference/ch_PP-OCRv3_rec_infer/
......
......@@ -9,7 +9,7 @@ inference:./deploy/cpp_infer/build/ppocr
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16
--precision:fp32
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
null:null
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_det.py
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16|int8
--precision:fp32
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
null:null
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_det.py
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16|int8
--precision:fp32
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
null:null
......
......@@ -9,7 +9,7 @@ inference:./deploy/cpp_infer/build/ppocr --rec_img_h=48 --rec_char_dict_path=./p
--cpu_threads:6
--rec_batch_num:6
--use_tensorrt:False
--precision:fp32|fp16
--precision:fp32
--rec_model_dir:
--image_dir:./inference/rec_inference/
null:null
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_rec.py --rec_image_shape="3,48,320"
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:False
--precision:fp32|int8
--precision:fp32
--rec_model_dir:
--image_dir:./inference/rec_inference
null:null
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_rec.py --rec_image_shape="3,48,320"
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:False
--precision:fp32|int8
--precision:fp32
--rec_model_dir:
--image_dir:./inference/rec_inference
null:null
......
......@@ -9,7 +9,7 @@ inference:./deploy/cpp_infer/build/ppocr --rec_char_dict_path=./ppocr/utils/ppoc
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16
--precision:fp32
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
--rec_model_dir:./inference/ch_ppocr_mobile_v2.0_rec_infer/
......
......@@ -9,7 +9,7 @@ inference:./deploy/cpp_infer/build/ppocr
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16
--precision:fp32
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
null:null
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_det.py
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16|int8
--precision:fp32
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
null:null
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_det.py
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16|int8
--precision:fp32
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
null:null
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_det.py
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16|int8
--precision:fp32
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
null:null
......
......@@ -9,7 +9,7 @@ inference:./deploy/cpp_infer/build/ppocr --rec_char_dict_path=./ppocr/utils/ppoc
--cpu_threads:6
--rec_batch_num:6
--use_tensorrt:False
--precision:fp32|fp16
--precision:fp32
--rec_model_dir:
--image_dir:./inference/rec_inference/
null:null
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_rec.py
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:False
--precision:fp32|int8
--precision:fp32
--rec_model_dir:
--image_dir:./inference/rec_inference
--save_log_path:./test/output/
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_rec.py
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|int8
--precision:fp32
--rec_model_dir:
--image_dir:./inference/rec_inference
null:null
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ppocr_ke
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:False
--precision:fp32|int8
--precision:fp32
--rec_model_dir:
--image_dir:./inference/rec_inference
--save_log_path:./test/output/
......
......@@ -9,7 +9,7 @@ inference:./deploy/cpp_infer/build/ppocr --rec_char_dict_path=./ppocr/utils/ppoc
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16
--precision:fp32
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
--rec_model_dir:./inference/ch_ppocr_server_v2.0_rec_infer/
......
......@@ -9,7 +9,7 @@ inference:./deploy/cpp_infer/build/ppocr
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16
--precision:fp32
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
null:null
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_det.py
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16|int8
--precision:fp32
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
--save_log_path:null
......
......@@ -9,7 +9,7 @@ inference:./deploy/cpp_infer/build/ppocr --rec_char_dict_path=./ppocr/utils/ppoc
--cpu_threads:6
--rec_batch_num:6
--use_tensorrt:False
--precision:fp32|fp16
--precision:fp32
--rec_model_dir:
--image_dir:./inference/rec_inference/
null:null
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_rec.py
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:False
--precision:fp32|int8
--precision:fp32
--rec_model_dir:
--image_dir:./inference/rec_inference
--save_log_path:./test/output/
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_det.py
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16|int8
--precision:fp32
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
null:null
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_det.py
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16|int8
--precision:fp32
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
--save_log_path:null
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_det.py
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16
--precision:fp32
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
--save_log_path:null
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_det.py
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16|int8
--precision:fp32
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
--save_log_path:null
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_det.py
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16|int8
--precision:fp32
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
--save_log_path:null
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_det.py
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16|int8
--precision:fp32
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
--save_log_path:null
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_det.py
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16|int8
--precision:fp32
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
--save_log_path:null
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_det.py
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|int8
--precision:fp32
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
null:null
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_det.py
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|int8
--precision:fp32
--det_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
null:null
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_e2e.py
--cpu_threads:6
--rec_batch_num:1
--use_tensorrt:False
--precision:fp32|fp16|int8
--precision:fp32
--e2e_model_dir:
--image_dir:./inference/ch_det_data_50/all-sum-510/
null:null
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/EN_symbo
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:False
--precision:fp32|int8
--precision:fp32
--rec_model_dir:
--image_dir:./inference/rec_inference
--save_log_path:./test/output/
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dic
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:False
--precision:fp32|int8
--precision:fp32
--rec_model_dir:
--image_dir:./inference/rec_inference
--save_log_path:./test/output/
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dic
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:False
--precision:fp32|int8
--precision:fp32
--rec_model_dir:
--image_dir:./inference/rec_inference
--save_log_path:./test/output/
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dic
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:False
--precision:fp32|int8
--precision:fp32
--rec_model_dir:
--image_dir:./inference/rec_inference
--save_log_path:./test/output/
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dic
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:False
--precision:fp32|int8
--precision:fp32
--rec_model_dir:
--image_dir:./inference/rec_inference
--save_log_path:./test/output/
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/dict90.t
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:False
--precision:fp32|int8
--precision:fp32
--rec_model_dir:
--image_dir:./inference/rec_inference
--save_log_path:./test/output/
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dic
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:False
--precision:fp32|int8
--precision:fp32
--rec_model_dir:
--image_dir:./inference/rec_inference
--save_log_path:./test/output/
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dic
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:False
--precision:fp32|int8
--precision:fp32
--rec_model_dir:
--image_dir:./inference/rec_inference
--save_log_path:./test/output/
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dic
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:False
--precision:fp32|int8
--precision:fp32
--rec_model_dir:
--image_dir:./inference/rec_inference
--save_log_path:./test/output/
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dic
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:False
--precision:fp32|int8
--precision:fp32
--rec_model_dir:
--image_dir:./inference/rec_inference
--save_log_path:./test/output/
......
......@@ -43,7 +43,7 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dic
--cpu_threads:6
--rec_batch_num:1|6
--use_tensorrt:False
--precision:fp32|int8
--precision:fp32
--rec_model_dir:
--image_dir:./inference/rec_inference
--save_log_path:./test/output/
......
......@@ -84,7 +84,7 @@ function func_cpp_inference(){
eval $command
last_status=${PIPESTATUS[0]}
eval "cat ${_save_log_path}"
status_check $last_status "${command}" "${status_log}"
status_check $last_status "${command}" "${status_log}" "${model_name}"
done
done
done
......@@ -117,7 +117,7 @@ function func_cpp_inference(){
eval $command
last_status=${PIPESTATUS[0]}
eval "cat ${_save_log_path}"
status_check $last_status "${command}" "${status_log}"
status_check $last_status "${command}" "${status_log}" "${model_name}"
done
done
......
......@@ -65,7 +65,7 @@ function func_paddle2onnx(){
trans_model_cmd="${padlle2onnx_cmd} ${set_dirname} ${set_model_filename} ${set_params_filename} ${set_save_model} ${set_opset_version} ${set_enable_onnx_checker}"
eval $trans_model_cmd
last_status=${PIPESTATUS[0]}
status_check $last_status "${trans_model_cmd}" "${status_log}"
status_check $last_status "${trans_model_cmd}" "${status_log}" "${model_name}"
# trans rec
set_dirname=$(func_set_params "--model_dir" "${rec_infer_model_dir_value}")
set_model_filename=$(func_set_params "${model_filename_key}" "${model_filename_value}")
......@@ -76,7 +76,7 @@ function func_paddle2onnx(){
trans_model_cmd="${padlle2onnx_cmd} ${set_dirname} ${set_model_filename} ${set_params_filename} ${set_save_model} ${set_opset_version} ${set_enable_onnx_checker}"
eval $trans_model_cmd
last_status=${PIPESTATUS[0]}
status_check $last_status "${trans_model_cmd}" "${status_log}"
status_check $last_status "${trans_model_cmd}" "${status_log}" "${model_name}"
elif [ ${model_name} = "ch_PP-OCRv2_det" ] || [ ${model_name} = "ch_PP-OCRv3_det" ] || [ ${model_name} = "ch_ppocr_mobile_v2.0_det" ] || [ ${model_name} = "ch_ppocr_server_v2.0_det" ]; then
# trans det
set_dirname=$(func_set_params "--model_dir" "${det_infer_model_dir_value}")
......@@ -88,7 +88,7 @@ function func_paddle2onnx(){
trans_model_cmd="${padlle2onnx_cmd} ${set_dirname} ${set_model_filename} ${set_params_filename} ${set_save_model} ${set_opset_version} ${set_enable_onnx_checker}"
eval $trans_model_cmd
last_status=${PIPESTATUS[0]}
status_check $last_status "${trans_model_cmd}" "${status_log}"
status_check $last_status "${trans_model_cmd}" "${status_log}" "${model_name}"
elif [ ${model_name} = "ch_PP-OCRv2_rec" ] || [ ${model_name} = "ch_PP-OCRv3_rec" ] || [ ${model_name} = "ch_ppocr_mobile_v2.0_rec" ] || [ ${model_name} = "ch_ppocr_server_v2.0_rec" ]; then
# trans rec
set_dirname=$(func_set_params "--model_dir" "${rec_infer_model_dir_value}")
......@@ -100,7 +100,7 @@ function func_paddle2onnx(){
trans_model_cmd="${padlle2onnx_cmd} ${set_dirname} ${set_model_filename} ${set_params_filename} ${set_save_model} ${set_opset_version} ${set_enable_onnx_checker}"
eval $trans_model_cmd
last_status=${PIPESTATUS[0]}
status_check $last_status "${trans_model_cmd}" "${status_log}"
status_check $last_status "${trans_model_cmd}" "${status_log}" "${model_name}"
fi
# python inference
......@@ -123,7 +123,7 @@ function func_paddle2onnx(){
eval $infer_model_cmd
last_status=${PIPESTATUS[0]}
eval "cat ${_save_log_path}"
status_check $last_status "${infer_model_cmd}" "${status_log}"
status_check $last_status "${infer_model_cmd}" "${status_log}" "${model_name}"
elif [ ${use_gpu} = "True" ] || [ ${use_gpu} = "gpu" ]; then
_save_log_path="${LOG_PATH}/paddle2onnx_infer_gpu.log"
set_gpu=$(func_set_params "${use_gpu_key}" "${use_gpu}")
......@@ -142,7 +142,7 @@ function func_paddle2onnx(){
eval $infer_model_cmd
last_status=${PIPESTATUS[0]}
eval "cat ${_save_log_path}"
status_check $last_status "${infer_model_cmd}" "${status_log}"
status_check $last_status "${infer_model_cmd}" "${status_log}" "${model_name}"
else
echo "Does not support hardware other than CPU and GPU Currently!"
fi
......
......@@ -169,7 +169,7 @@ function func_inference(){
eval $command
last_status=${PIPESTATUS[0]}
eval "cat ${_save_log_path}"
status_check $last_status "${command}" "${status_log}"
status_check $last_status "${command}" "${status_log}" "${model_name}"
done
done
done
......@@ -200,7 +200,7 @@ function func_inference(){
eval $command
last_status=${PIPESTATUS[0]}
eval "cat ${_save_log_path}"
status_check $last_status "${command}" "${status_log}"
status_check $last_status "${command}" "${status_log}" "${model_name}"
done
done
......@@ -240,7 +240,7 @@ if [ ${MODE} = "whole_infer" ] || [ ${MODE} = "klquant_whole_infer" ]; then
echo $export_cmd
eval $export_cmd
status_export=$?
status_check $status_export "${export_cmd}" "${status_log}"
status_check $status_export "${export_cmd}" "${status_log}" "${model_name}"
else
save_infer_dir=${infer_model}
fi
......@@ -337,7 +337,7 @@ else
fi
# run train
eval $cmd
status_check $? "${cmd}" "${status_log}"
status_check $? "${cmd}" "${status_log}" "${model_name}"
set_eval_pretrain=$(func_set_params "${pretrain_model_key}" "${save_log}/${train_model_name}")
......@@ -347,7 +347,7 @@ else
set_eval_params1=$(func_set_params "${eval_key1}" "${eval_value1}")
eval_cmd="${python} ${eval_py} ${set_eval_pretrain} ${set_use_gpu} ${set_eval_params1}"
eval $eval_cmd
status_check $? "${eval_cmd}" "${status_log}"
status_check $? "${eval_cmd}" "${status_log}" "${model_name}"
fi
# run export model
if [ ${run_export} != "null" ]; then
......@@ -357,7 +357,7 @@ else
set_save_infer_key=$(func_set_params "${save_infer_key}" "${save_infer_path}")
export_cmd="${python} ${run_export} ${set_export_weight} ${set_save_infer_key}"
eval $export_cmd
status_check $? "${export_cmd}" "${status_log}"
status_check $? "${export_cmd}" "${status_log}" "${model_name}"
#run inference
eval $env
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册