opensource/PaddleOCR
3
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2026-09-21 13:23:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

update for tipc

Browse Source
This commit is contained in:
andyjpaddle
2022-05-27 02:38:23 +00:00
parent b7656e684e
commit 8ffef11f5e
52 changed files with 63 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test_tipc/configs/ch_PP-OCRv2/model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt
+1 -1
View File
@@ -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/
test_tipc/configs/ch_PP-OCRv2_det/model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt
+1 -1
View File
@@ -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
test_tipc/configs/ch_PP-OCRv2_det/train_infer_python.txt
+1 -1
View File
@@ -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
test_tipc/configs/ch_PP-OCRv2_det_PACT/train_infer_python.txt
+1 -1
View File
@@ -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
test_tipc/configs/ch_PP-OCRv2_rec/model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt
+1 -1
View File
@@ -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
test_tipc/configs/ch_PP-OCRv2_rec/train_infer_python.txt
+1 -1
View File
@@ -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
test_tipc/configs/ch_PP-OCRv2_rec_PACT/train_infer_python.txt
+1 -1
View File
@@ -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
test_tipc/configs/ch_PP-OCRv3/model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt
+1 -1
View File
@@ -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/
test_tipc/configs/ch_PP-OCRv3_det/model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt
+1 -1
View File
@@ -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
test_tipc/configs/ch_PP-OCRv3_det/train_infer_python.txt
+1 -1
View File
@@ -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
test_tipc/configs/ch_PP-OCRv3_det_PACT/train_infer_python.txt
+1 -1
View File
@@ -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
test_tipc/configs/ch_PP-OCRv3_rec/model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt
+1 -1
View File
@@ -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
test_tipc/configs/ch_PP-OCRv3_rec/train_infer_python.txt
+1 -1
View File
@@ -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
test_tipc/configs/ch_PP-OCRv3_rec_PACT/train_infer_python.txt
+1 -1
View File
@@ -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
test_tipc/configs/ch_ppocr_mobile_v2.0/model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt
+1 -1
View File
@@ -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/
test_tipc/configs/ch_ppocr_mobile_v2.0_det/model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt
+1 -1
View File
@@ -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
test_tipc/configs/ch_ppocr_mobile_v2.0_det/train_infer_python.txt
+1 -1
View File
@@ -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
test_tipc/configs/ch_ppocr_mobile_v2.0_det_FPGM/train_infer_python.txt
+1 -1
View File
@@ -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
test_tipc/configs/ch_ppocr_mobile_v2.0_det_PACT/train_infer_python.txt
+1 -1
View File
@@ -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
test_tipc/configs/ch_ppocr_mobile_v2.0_rec/model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt
+1 -1
View File
@@ -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
test_tipc/configs/ch_ppocr_mobile_v2.0_rec/train_infer_python.txt
+1 -1
View File
@@ -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/
test_tipc/configs/ch_ppocr_mobile_v2.0_rec_FPGM/train_infer_python.txt
+1 -1
View File
@@ -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
test_tipc/configs/ch_ppocr_mobile_v2.0_rec_PACT/train_infer_python.txt
+1 -1
View File
@@ -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/
test_tipc/configs/ch_ppocr_server_v2.0/model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt
+1 -1
View File
@@ -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/
test_tipc/configs/ch_ppocr_server_v2.0_det/model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt
+1 -1
View File
@@ -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
test_tipc/configs/ch_ppocr_server_v2.0_det/train_infer_python.txt
+1 -1
View File
@@ -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
test_tipc/configs/ch_ppocr_server_v2.0_rec/model_linux_gpu_normal_normal_infer_cpp_linux_gpu_cpu.txt
+1 -1
View File
@@ -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
test_tipc/configs/ch_ppocr_server_v2.0_rec/train_infer_python.txt
+1 -1
View File
@@ -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/
test_tipc/configs/det_mv3_db_v2_0/train_infer_python.txt
+1 -1
View File
@@ -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
test_tipc/configs/det_mv3_east_v2.0/train_infer_python.txt
+1 -1
View File
@@ -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
test_tipc/configs/det_mv3_pse_v2.0/train_infer_python.txt
+1 -1
View File
@@ -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
test_tipc/configs/det_r18_vd_db_v2_0/train_infer_python.txt
+1 -1
View File
@@ -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
test_tipc/configs/det_r50_db_v2.0/train_infer_python.txt
+1 -1
View File
@@ -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
test_tipc/configs/det_r50_vd_east_v2_0/train_infer_python.txt
+1 -1
View File
@@ -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
test_tipc/configs/det_r50_vd_pse_v2_0/train_infer_python.txt
+1 -1
View File
@@ -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
test_tipc/configs/det_r50_vd_sast_icdar15_v2.0/train_infer_python.txt
+1 -1
View File
@@ -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
test_tipc/configs/det_r50_vd_sast_totaltext_v2.0/train_infer_python.txt
+1 -1
View File
@@ -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
test_tipc/configs/en_server_pgnetA/train_infer_python.txt
+1 -1
View File
@@ -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
test_tipc/configs/rec_mtb_nrtr/train_infer_python.txt
+1 -1
View File
@@ -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/
test_tipc/configs/rec_mv3_none_bilstm_ctc_v2.0/train_infer_python.txt
+1 -1
View File
@@ -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/
test_tipc/configs/rec_mv3_none_none_ctc_v2.0/train_infer_python.txt
+1 -1
View File
@@ -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/
test_tipc/configs/rec_mv3_tps_bilstm_att_v2.0/train_infer_python.txt
+1 -1
View File
@@ -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/
test_tipc/configs/rec_mv3_tps_bilstm_ctc_v2.0/train_infer_python.txt
+1 -1
View File
@@ -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/
test_tipc/configs/rec_r31_sar/train_infer_python.txt
+1 -1
View File
@@ -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/
test_tipc/configs/rec_r34_vd_none_bilstm_ctc_v2.0/train_infer_python.txt
+1 -1
View File
@@ -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/
test_tipc/configs/rec_r34_vd_none_none_ctc_v2.0/train_infer_python.txt
+1 -1
View File
@@ -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/
test_tipc/configs/rec_r34_vd_tps_bilstm_att_v2.0/train_infer_python.txt
+1 -1
View File
@@ -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/
test_tipc/configs/rec_r34_vd_tps_bilstm_ctc_v2.0/train_infer_python.txt
+1 -1
View File
@@ -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/
test_tipc/configs/rec_r50_fpn_vd_none_srn/train_infer_python.txt
+1 -1
View File
@@ -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/
test_tipc/test_inference_cpp.sh
+2 -2
View File
@@ -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
test_tipc/test_paddle2onnx.sh
+6 -6
View File
@@ -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
test_tipc/test_train_inference_python.sh
+6 -6
View File
@@ -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