From 40d235f54ea49b328838125edf540b08df98f7df Mon Sep 17 00:00:00 2001 From: andyjpaddle Date: Fri, 4 Mar 2022 07:44:27 +0000 Subject: [PATCH] add infer benchmark params for rec model --- test_tipc/configs/ch_PP-OCRv2_rec/train_infer_python.txt | 4 ++-- test_tipc/configs/ch_PP-OCRv2_rec_PACT/train_infer_python.txt | 4 ++-- .../configs/ch_ppocr_mobile_v2.0_rec/train_infer_python.txt | 2 ++ .../ch_ppocr_mobile_v2.0_rec_FPGM/train_infer_python.txt | 4 +++- .../ch_ppocr_mobile_v2.0_rec_PACT/train_infer_python.txt | 4 +++- .../configs/ch_ppocr_server_v2.0_rec/train_infer_python.txt | 2 ++ test_tipc/configs/rec_mtb_nrtr/train_infer_python.txt | 3 ++- .../rec_mv3_none_bilstm_ctc_v2.0/train_infer_python.txt | 2 ++ .../configs/rec_mv3_none_none_ctc_v2.0/train_infer_python.txt | 2 ++ .../rec_mv3_tps_bilstm_att_v2.0/train_infer_python.txt | 3 ++- .../rec_mv3_tps_bilstm_ctc_v2.0/train_infer_python.txt | 2 ++ test_tipc/configs/rec_r31_sar/train_infer_python.txt | 3 ++- .../rec_r34_vd_none_bilstm_ctc_v2.0/train_infer_python.txt | 2 ++ .../rec_r34_vd_none_none_ctc_v2.0/train_infer_python.txt | 2 ++ .../rec_r34_vd_tps_bilstm_att_v2.0/train_infer_python.txt | 3 ++- .../rec_r34_vd_tps_bilstm_ctc_v2.0/train_infer_python.txt | 2 ++ .../configs/rec_r50_fpn_vd_none_srn/train_infer_python.txt | 3 ++- 17 files changed, 36 insertions(+), 11 deletions(-) diff --git a/test_tipc/configs/ch_PP-OCRv2_rec/train_infer_python.txt b/test_tipc/configs/ch_PP-OCRv2_rec/train_infer_python.txt index b2de2a5e5..188eb3ccc 100644 --- a/test_tipc/configs/ch_PP-OCRv2_rec/train_infer_python.txt +++ b/test_tipc/configs/ch_PP-OCRv2_rec/train_infer_python.txt @@ -49,5 +49,5 @@ inference:tools/infer/predict_rec.py null:null --benchmark:True null:null - - +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,32,320]}] diff --git a/test_tipc/configs/ch_PP-OCRv2_rec_PACT/train_infer_python.txt b/test_tipc/configs/ch_PP-OCRv2_rec_PACT/train_infer_python.txt index 9102382fd..03d749f55 100644 --- a/test_tipc/configs/ch_PP-OCRv2_rec_PACT/train_infer_python.txt +++ b/test_tipc/configs/ch_PP-OCRv2_rec_PACT/train_infer_python.txt @@ -49,5 +49,5 @@ inference:tools/infer/predict_rec.py null:null --benchmark:True null:null - - +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,32,320]}] diff --git a/test_tipc/configs/ch_ppocr_mobile_v2.0_rec/train_infer_python.txt b/test_tipc/configs/ch_ppocr_mobile_v2.0_rec/train_infer_python.txt index 63d1f0583..5086f80d7 100644 --- a/test_tipc/configs/ch_ppocr_mobile_v2.0_rec/train_infer_python.txt +++ b/test_tipc/configs/ch_ppocr_mobile_v2.0_rec/train_infer_python.txt @@ -49,3 +49,5 @@ inference:tools/infer/predict_rec.py --save_log_path:./test/output/ --benchmark:True null:null +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,32,100]}] diff --git a/test_tipc/configs/ch_ppocr_mobile_v2.0_rec_FPGM/train_infer_python.txt b/test_tipc/configs/ch_ppocr_mobile_v2.0_rec_FPGM/train_infer_python.txt index c5dfa46f3..77494ac34 100644 --- a/test_tipc/configs/ch_ppocr_mobile_v2.0_rec_FPGM/train_infer_python.txt +++ b/test_tipc/configs/ch_ppocr_mobile_v2.0_rec_FPGM/train_infer_python.txt @@ -48,4 +48,6 @@ inference:tools/infer/predict_rec.py --image_dir:./inference/rec_inference null:null --benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,32,320]}] \ No newline at end of file diff --git a/test_tipc/configs/ch_ppocr_mobile_v2.0_rec_PACT/train_infer_python.txt b/test_tipc/configs/ch_ppocr_mobile_v2.0_rec_PACT/train_infer_python.txt index afbf2ef5e..94909ec34 100644 --- a/test_tipc/configs/ch_ppocr_mobile_v2.0_rec_PACT/train_infer_python.txt +++ b/test_tipc/configs/ch_ppocr_mobile_v2.0_rec_PACT/train_infer_python.txt @@ -48,4 +48,6 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ppocr_ke --image_dir:./inference/rec_inference --save_log_path:./test/output/ --benchmark:True -null:null \ No newline at end of file +null:null +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,32,320]}] diff --git a/test_tipc/configs/ch_ppocr_server_v2.0_rec/train_infer_python.txt b/test_tipc/configs/ch_ppocr_server_v2.0_rec/train_infer_python.txt index c42edbee4..78a046c50 100644 --- a/test_tipc/configs/ch_ppocr_server_v2.0_rec/train_infer_python.txt +++ b/test_tipc/configs/ch_ppocr_server_v2.0_rec/train_infer_python.txt @@ -49,3 +49,5 @@ inference:tools/infer/predict_rec.py --save_log_path:./test/output/ --benchmark:True null:null +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,32,100]}] diff --git a/test_tipc/configs/rec_mtb_nrtr/train_infer_python.txt b/test_tipc/configs/rec_mtb_nrtr/train_infer_python.txt index 2adca464a..de6de5a0c 100644 --- a/test_tipc/configs/rec_mtb_nrtr/train_infer_python.txt +++ b/test_tipc/configs/rec_mtb_nrtr/train_infer_python.txt @@ -49,4 +49,5 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/EN_symbo --save_log_path:./test/output/ --benchmark:True null:null - +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[1,32,100]}] diff --git a/test_tipc/configs/rec_mv3_none_bilstm_ctc_v2.0/train_infer_python.txt b/test_tipc/configs/rec_mv3_none_bilstm_ctc_v2.0/train_infer_python.txt index ac565d8c5..e67dd1509 100644 --- a/test_tipc/configs/rec_mv3_none_bilstm_ctc_v2.0/train_infer_python.txt +++ b/test_tipc/configs/rec_mv3_none_bilstm_ctc_v2.0/train_infer_python.txt @@ -49,3 +49,5 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dic --save_log_path:./test/output/ --benchmark:True null:null +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,32,100]}] \ No newline at end of file diff --git a/test_tipc/configs/rec_mv3_none_none_ctc_v2.0/train_infer_python.txt b/test_tipc/configs/rec_mv3_none_none_ctc_v2.0/train_infer_python.txt index 947399a83..aa3e88d28 100644 --- a/test_tipc/configs/rec_mv3_none_none_ctc_v2.0/train_infer_python.txt +++ b/test_tipc/configs/rec_mv3_none_none_ctc_v2.0/train_infer_python.txt @@ -49,3 +49,5 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dic --save_log_path:./test/output/ --benchmark:True null:null +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,32,100]}] \ No newline at end of file diff --git a/test_tipc/configs/rec_mv3_tps_bilstm_att_v2.0/train_infer_python.txt b/test_tipc/configs/rec_mv3_tps_bilstm_att_v2.0/train_infer_python.txt index 5fcfeee5e..32df669f9 100644 --- a/test_tipc/configs/rec_mv3_tps_bilstm_att_v2.0/train_infer_python.txt +++ b/test_tipc/configs/rec_mv3_tps_bilstm_att_v2.0/train_infer_python.txt @@ -49,4 +49,5 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dic --save_log_path:./test/output/ --benchmark:True null:null - +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,32,100]}] diff --git a/test_tipc/configs/rec_mv3_tps_bilstm_ctc_v2.0/train_infer_python.txt b/test_tipc/configs/rec_mv3_tps_bilstm_ctc_v2.0/train_infer_python.txt index ac3fce614..7a3096eb1 100644 --- a/test_tipc/configs/rec_mv3_tps_bilstm_ctc_v2.0/train_infer_python.txt +++ b/test_tipc/configs/rec_mv3_tps_bilstm_ctc_v2.0/train_infer_python.txt @@ -49,3 +49,5 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dic --save_log_path:./test/output/ --benchmark:True null:null +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,32,100]}] diff --git a/test_tipc/configs/rec_r31_sar/train_infer_python.txt b/test_tipc/configs/rec_r31_sar/train_infer_python.txt index e4d782524..c5018500f 100644 --- a/test_tipc/configs/rec_r31_sar/train_infer_python.txt +++ b/test_tipc/configs/rec_r31_sar/train_infer_python.txt @@ -49,4 +49,5 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/dict90.t --save_log_path:./test/output/ --benchmark:True null:null - +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,48,48,160]}] diff --git a/test_tipc/configs/rec_r34_vd_none_bilstm_ctc_v2.0/train_infer_python.txt b/test_tipc/configs/rec_r34_vd_none_bilstm_ctc_v2.0/train_infer_python.txt index 99f868725..02cea56fb 100644 --- a/test_tipc/configs/rec_r34_vd_none_bilstm_ctc_v2.0/train_infer_python.txt +++ b/test_tipc/configs/rec_r34_vd_none_bilstm_ctc_v2.0/train_infer_python.txt @@ -49,3 +49,5 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dic --save_log_path:./test/output/ --benchmark:True null:null +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,32,100]}] \ No newline at end of file diff --git a/test_tipc/configs/rec_r34_vd_none_none_ctc_v2.0/train_infer_python.txt b/test_tipc/configs/rec_r34_vd_none_none_ctc_v2.0/train_infer_python.txt index fb1ece49f..5e7c1d343 100644 --- a/test_tipc/configs/rec_r34_vd_none_none_ctc_v2.0/train_infer_python.txt +++ b/test_tipc/configs/rec_r34_vd_none_none_ctc_v2.0/train_infer_python.txt @@ -49,3 +49,5 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dic --save_log_path:./test/output/ --benchmark:True null:null +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,32,100]}] \ No newline at end of file diff --git a/test_tipc/configs/rec_r34_vd_tps_bilstm_att_v2.0/train_infer_python.txt b/test_tipc/configs/rec_r34_vd_tps_bilstm_att_v2.0/train_infer_python.txt index acc9749f0..9cee5d0b7 100644 --- a/test_tipc/configs/rec_r34_vd_tps_bilstm_att_v2.0/train_infer_python.txt +++ b/test_tipc/configs/rec_r34_vd_tps_bilstm_att_v2.0/train_infer_python.txt @@ -49,4 +49,5 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dic --save_log_path:./test/output/ --benchmark:True null:null - +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,32,100]}] diff --git a/test_tipc/configs/rec_r34_vd_tps_bilstm_ctc_v2.0/train_infer_python.txt b/test_tipc/configs/rec_r34_vd_tps_bilstm_ctc_v2.0/train_infer_python.txt index d11850528..5b5ba0fd0 100644 --- a/test_tipc/configs/rec_r34_vd_tps_bilstm_ctc_v2.0/train_infer_python.txt +++ b/test_tipc/configs/rec_r34_vd_tps_bilstm_ctc_v2.0/train_infer_python.txt @@ -49,3 +49,5 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dic --save_log_path:./test/output/ --benchmark:True null:null +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[3,32,100]}] diff --git a/test_tipc/configs/rec_r50_fpn_vd_none_srn/train_infer_python.txt b/test_tipc/configs/rec_r50_fpn_vd_none_srn/train_infer_python.txt index fb135df60..187c1cc13 100644 --- a/test_tipc/configs/rec_r50_fpn_vd_none_srn/train_infer_python.txt +++ b/test_tipc/configs/rec_r50_fpn_vd_none_srn/train_infer_python.txt @@ -49,4 +49,5 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dic --save_log_path:./test/output/ --benchmark:True null:null - +===========================infer_benchmark_params========================== +random_infer_input:[{float32,[1,64,256]}]