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 f35bb3876737606f58949968d5b43f3bd796be30..f10985a91902716968660af5188473e4f1a7ae3d 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 @@ -37,13 +37,13 @@ export2:null infer_model:null infer_export:tools/export_model.py -c test_tipc/configs/ch_ppocr_server_v2.0_rec/rec_icdar15_train.yml -o infer_quant:False -inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dict.txt --rec_image_shape="3,32,100" +inference:tools/infer/predict_rec.py --use_gpu:True|False --enable_mkldnn:True|False --cpu_threads:1|6 --rec_batch_num:1|6 --use_tensorrt:True|False ---precision:fp32|fp16|int8 +--precision:fp32|int8 --rec_model_dir: --image_dir:./inference/rec_inference --save_log_path:./test/output/ 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 698c14ee66914bea7a6926650234ad7a979d01b4..fdc39f1b851a4e05735744f878917a3dfcc1d405 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 @@ -43,7 +43,7 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dic --cpu_threads:1|6 --rec_batch_num:1|6 --use_tensorrt:True|False ---precision:fp32|fp16|int8 +--precision:fp32|int8 --rec_model_dir: --image_dir:./inference/rec_inference --save_log_path:./test/output/ 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 53250c57e49967b9816b62dae07c64c0d22d58af..9810689679903b4cedff253834a1a999c4e8a5f8 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 @@ -43,7 +43,7 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dic --cpu_threads:1|6 --rec_batch_num:1|6 --use_tensorrt:True|False ---precision:fp32|fp16|int8 +--precision:fp32|int8 --rec_model_dir: --image_dir:./inference/rec_inference --save_log_path:./test/output/ 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 5de24ae5835f91ee2b4a6d7f816197ce694002f6..33700ad696394ad9404a5424cddf93608220917a 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 @@ -43,7 +43,7 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dic --cpu_threads:1|6 --rec_batch_num:1|6 --use_tensorrt:True|False ---precision:fp32|fp16|int8 +--precision:fp32|int8 --rec_model_dir: --image_dir:./inference/rec_inference --save_log_path:./test/output/ 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 baf306e1897c9f0f65ebe45747738b18173dd286..857a212fe6f5e0bd9612b55841e748e6b4409061 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 @@ -43,7 +43,7 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dic --cpu_threads:1|6 --rec_batch_num:1|6 --use_tensorrt:True|False ---precision:fp32|fp16|int8 +--precision:fp32|int8 --rec_model_dir: --image_dir:./inference/rec_inference --save_log_path:./test/output/ 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 0d54bd5647d9cd4d8b1ffec5b2baa99874cff7f6..85804b7453729dad6d2e87d0efd1a053dd9a0aac 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 @@ -43,7 +43,7 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dic --cpu_threads:1|6 --rec_batch_num:1|6 --use_tensorrt:True|False ---precision:fp32|fp16|int8 +--precision:fp32|int8 --rec_model_dir: --image_dir:./inference/rec_inference --save_log_path:./test/output/ 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 666c7d159b4cecb262197b3373ba3df125037d67..bb49ae5977208b2921f4a825b62afa7935f572f1 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 @@ -43,7 +43,7 @@ inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dic --cpu_threads:1|6 --rec_batch_num:1|6 --use_tensorrt:True|False ---precision:fp32|fp16|int8 +--precision:fp32|int8 --rec_model_dir: --image_dir:./inference/rec_inference --save_log_path:./test/output/ diff --git a/test_tipc/configs/rec_resnet_stn_bilstm_att_v2.0/rec_icdar15_train.yml b/test_tipc/configs/rec_resnet_stn_bilstm_att_v2.0/rec_icdar15_train.yml deleted file mode 100644 index acac033258f8732bb1e015d967d498cc9bf2ca51..0000000000000000000000000000000000000000 --- a/test_tipc/configs/rec_resnet_stn_bilstm_att_v2.0/rec_icdar15_train.yml +++ /dev/null @@ -1,110 +0,0 @@ -Global: - use_gpu: True - epoch_num: 400 - log_smooth_window: 20 - print_batch_step: 10 - save_model_dir: ./output/rec/seed - save_epoch_step: 3 - # evaluation is run every 5000 iterations after the 4000th iteration - eval_batch_step: [0, 2000] - cal_metric_during_train: True - pretrained_model: - checkpoints: - save_inference_dir: - use_visualdl: False - infer_img: doc/imgs_words_en/word_10.png - # for data or label process - character_dict_path: ppocr/utils/EN_symbol_dict.txt - max_text_length: 100 - infer_mode: False - use_space_char: False - save_res_path: ./output/rec/predicts_seed.txt - - -Optimizer: - name: Adadelta - weight_deacy: 0.0 - momentum: 0.9 - lr: - name: Piecewise - decay_epochs: [4,5,8] - values: [1.0, 0.1, 0.01] - regularizer: - name: 'L2' - factor: 2.0e-05 - - -Architecture: - model_type: rec - algorithm: SEED - Transform: - name: STN_ON - tps_inputsize: [32, 64] - tps_outputsize: [32, 100] - num_control_points: 20 - tps_margins: [0.05,0.05] - stn_activation: none - Backbone: - name: ResNet_ASTER - Head: - name: AsterHead # AttentionHead - sDim: 512 - attDim: 512 - max_len_labels: 100 - -Loss: - name: AsterLoss - -PostProcess: - name: SEEDLabelDecode - -Metric: - name: RecMetric - main_indicator: acc - is_filter: True - -Train: - dataset: - name: SimpleDataSet - data_dir: ./train_data/ic15_data/ - label_file_list: ["./train_data/ic15_data/rec_gt_train.txt"] - transforms: - - Fasttext: - path: "./cc.en.300.bin" - - DecodeImage: # load image - img_mode: BGR - channel_first: False - - SEEDLabelEncode: # Class handling label - - RecResizeImg: - character_type: en - image_shape: [3, 64, 256] - padding: False - - KeepKeys: - keep_keys: ['image', 'label', 'length', 'fast_label'] # dataloader will return list in this order - loader: - shuffle: True - batch_size_per_card: 256 - drop_last: True - num_workers: 6 - -Eval: - dataset: - name: SimpleDataSet - data_dir: ./train_data/ic15_data - label_file_list: ["./train_data/ic15_data/rec_gt_test.txt"] - transforms: - - DecodeImage: # load image - img_mode: BGR - channel_first: False - - SEEDLabelEncode: # Class handling label - - RecResizeImg: - character_type: en - image_shape: [3, 64, 256] - padding: False - - KeepKeys: - keep_keys: ['image', 'label', 'length'] # dataloader will return list in this order - loader: - shuffle: False - drop_last: True - batch_size_per_card: 256 - num_workers: 4 diff --git a/test_tipc/configs/rec_resnet_stn_bilstm_att_v2.0/train_infer_python.txt b/test_tipc/configs/rec_resnet_stn_bilstm_att_v2.0/train_infer_python.txt deleted file mode 100644 index 9049dd0062cdade523baaa94d265adaf506d32f4..0000000000000000000000000000000000000000 --- a/test_tipc/configs/rec_resnet_stn_bilstm_att_v2.0/train_infer_python.txt +++ /dev/null @@ -1,51 +0,0 @@ -===========================train_params=========================== -model_name:rec_resnet_stn_bilstm_att_v2.0 -python:python3.7 -gpu_list:0|0,1 -Global.use_gpu:True|True -Global.auto_cast:null -Global.epoch_num:lite_train_lite_infer=5|whole_train_whole_infer=100 -Global.save_model_dir:./output/ -Train.loader.batch_size_per_card:lite_train_lite_infer=128|whole_train_whole_infer=128 -Global.pretrained_model:null -train_model_name:latest -train_infer_img_dir:./inference/rec_inference -null:null -## -trainer:norm_train -norm_train:tools/train.py -c test_tipc/configs/rec_resnet_stn_bilstm_att_v2.0/rec_icdar15_train.yml -o -pact_train:null -fpgm_train:null -distill_train:null -null:null -null:null -## -===========================eval_params=========================== -eval:tools/eval.py -c test_tipc/configs/rec_resnet_stn_bilstm_att_v2.0/rec_icdar15_train.yml -o -null:null -## -===========================infer_params=========================== -Global.save_inference_dir:./output/ -Global.pretrained_model: -norm_export:tools/export_model.py -c test_tipc/configs/rec_resnet_stn_bilstm_att_v2.0/rec_icdar15_train.yml -o -quant_export:null -fpgm_export:null -distill_export:null -export1:null -export2:null -## -infer_model:null -infer_export:tools/export_model.py -c test_tipc/configs/rec_resnet_stn_bilstm_att_v2.0/rec_icdar15_train.yml -o -infer_quant:False -inference:tools/infer/predict_rec.py --rec_char_dict_path=./ppocr/utils/ic15_dict.txt --rec_image_shape="3,32,100" ---use_gpu:True|False ---enable_mkldnn:True|False ---cpu_threads:1|6 ---rec_batch_num:1|6 ---use_tensorrt:True|False ---precision:fp32|fp16|int8 ---rec_model_dir: ---image_dir:./inference/rec_inference ---save_log_path:./test/output/ ---benchmark:True -null:null