forked from PaddlePaddle/PaddleOCR
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
update slanet and vi_layoutxlm_ser tipc config (PaddlePaddle#9838)
* update table_mv3 tipc config * update slanet tipc config * update num_workers of vi_layoutxlm_ser * add ch_PP-OCRv4_mobile_rec * add PP-OCRv4 * update slanet tipc config * change python3.7 to python * update vi_layoutxlm_ser config * update slanet tipc config * update v4 rec config
- Loading branch information
Showing
8 changed files
with
274 additions
and
153 deletions.
There are no files selected for viewing
61 changes: 61 additions & 0 deletions
61
test_tipc/configs/ch_PP-OCRv4_mobile_det/train_infer_python.txt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,61 @@ | ||
| ===========================train_params=========================== | ||
| model_name:ch_PP-OCRv4_mobile_det | ||
| python:python | ||
| gpu_list:0|0,1 | ||
| Global.use_gpu:True|True | ||
| Global.auto_cast:fp32 | ||
| Global.epoch_num:lite_train_lite_infer=1|whole_train_whole_infer=50 | ||
| Global.save_model_dir:./output/ | ||
| Train.loader.batch_size_per_card:lite_train_lite_infer=2|whole_train_whole_infer=4 | ||
| Global.pretrained_model:pretrain_models/PPLCNetV3_x0_75_ocr_det.pdparams | ||
| train_model_name:latest | ||
| train_infer_img_dir:./train_data/icdar2015/text_localization/ch4_test_images/ | ||
| null:null | ||
| ## | ||
| trainer:norm_train | ||
| norm_train:tools/train.py -c configs/det/ch_PP-OCRv4/ch_PP-OCRv4_det_student.yml -o Global.print_batch_step=1 Train.loader.shuffle=false Global.eval_batch_step=[4000,400] | ||
| pact_train:null | ||
| fpgm_train:null | ||
| distill_train:null | ||
| to_static_train:Global.to_static=true | ||
| null:null | ||
| ## | ||
| ===========================eval_params=========================== | ||
| eval:null | ||
| null:null | ||
| ## | ||
| ===========================infer_params=========================== | ||
| Global.save_inference_dir:./output/ | ||
| Global.checkpoints: | ||
| norm_export:tools/export_model.py -c configs/det/ch_PP-OCRv4/ch_PP-OCRv4_det_student.yml -o | ||
| quant_export:null | ||
| fpgm_export: | ||
| distill_export:null | ||
| export1:null | ||
| export2:null | ||
| inference_dir:Student | ||
| infer_model:./inference/ch_PP-OCRv3_det_infer/ | ||
| infer_export:null | ||
| infer_quant:False | ||
| inference:tools/infer/predict_det.py | ||
| --use_gpu:True|False | ||
| --enable_mkldnn:False | ||
| --cpu_threads:6 | ||
| --rec_batch_num:1 | ||
| --use_tensorrt:False | ||
| --precision:fp32 | ||
| --det_model_dir: | ||
| --image_dir:./inference/ch_det_data_50/all-sum-510/ | ||
| null:null | ||
| --benchmark:True | ||
| null:null | ||
| ===========================infer_benchmark_params========================== | ||
| random_infer_input:[{float32,[3,640,640]}];[{float32,[3,960,960]}] | ||
| ===========================train_benchmark_params========================== | ||
| batch_size:8 | ||
| fp_items:fp32|fp16 | ||
| epoch:2 | ||
| --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile | ||
| flags:FLAGS_eager_delete_tensor_gb=0.0;FLAGS_fraction_of_gpu_memory_to_use=0.98;FLAGS_conv_workspace_size_limit=4096 | ||
| ===========================to_static_train_benchmark_params=========================== | ||
| to_static_train:Global.to_static=true |
61 changes: 61 additions & 0 deletions
61
test_tipc/configs/ch_PP-OCRv4_mobile_rec/train_infer_python.txt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,61 @@ | ||
| ===========================train_params=========================== | ||
| model_name:ch_PP-OCRv4_mobile_rec | ||
| python:python | ||
| gpu_list:0 | ||
| Global.use_gpu:True|True | ||
| Global.auto_cast:fp32 | ||
| Global.epoch_num:lite_train_lite_infer=3|whole_train_whole_infer=50 | ||
| Global.save_model_dir:./output/ | ||
| Train.loader.batch_size_per_card:lite_train_lite_infer=16|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 configs/rec/PP-OCRv4/ch_PP-OCRv4_rec.yml -o Global.cal_metric_during_train=False Global.print_batch_step=1 Train.loader.shuffle=false Train.dataset.data_dir=./train_data/ic15_data Train.dataset.label_file_list=[./train_data/ic15_data/rec_gt_train.txt] Eval.dataset.data_dir=./train_data/ic15_data Eval.dataset.label_file_list=[./train_data/ic15_data/rec_gt_test.txt] Train.loader.num_workers=16 Eval.loader.num_workers=16 | ||
| pact_train:null | ||
| fpgm_train:null | ||
| distill_train:null | ||
| to_static_train:Global.to_static=true | ||
| null:null | ||
| ## | ||
| ===========================eval_params=========================== | ||
| eval:null | ||
| null:null | ||
| ## | ||
| ===========================infer_params=========================== | ||
| Global.save_inference_dir:./output/ | ||
| Global.checkpoints: | ||
| norm_export:tools/export_model.py -c configs/rec/PP-OCRv4/ch_PP-OCRv4_rec.yml -o | ||
| quant_export: | ||
| fpgm_export: | ||
| distill_export:null | ||
| export1:null | ||
| export2:null | ||
| ## | ||
| infer_model:./inference/ch_PP-OCRv4_rec_infer | ||
| infer_export:null | ||
| infer_quant:False | ||
| inference:tools/infer/predict_rec.py --rec_image_shape="3,48,320" | ||
| --use_gpu:True|False | ||
| --enable_mkldnn:False | ||
| --cpu_threads:6 | ||
| --rec_batch_num:1 | ||
| --use_tensorrt:False | ||
| --precision:fp32 | ||
| --rec_model_dir: | ||
| --image_dir:./inference/rec_inference | ||
| null:null | ||
| --benchmark:True | ||
| null:null | ||
| ===========================infer_benchmark_params========================== | ||
| random_infer_input:[{float32,[3,48,320]}] | ||
| ===========================train_benchmark_params========================== | ||
| batch_size:128 | ||
| fp_items:fp32|fp16 | ||
| epoch:1 | ||
| --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile | ||
| flags:FLAGS_eager_delete_tensor_gb=0.0;FLAGS_fraction_of_gpu_memory_to_use=0.98;FLAGS_conv_workspace_size_limit=4096 | ||
| ===========================to_static_train_benchmark_params=========================== | ||
| to_static_train:Global.to_static=true |
61 changes: 61 additions & 0 deletions
61
test_tipc/configs/ch_PP-OCRv4_server_det/train_infer_python.txt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,61 @@ | ||
| ===========================train_params=========================== | ||
| model_name:ch_PP-OCRv4_server_det | ||
| python:python | ||
| gpu_list:0|0,1 | ||
| Global.use_gpu:True|True | ||
| Global.auto_cast:fp32 | ||
| Global.epoch_num:lite_train_lite_infer=1|whole_train_whole_infer=50 | ||
| Global.save_model_dir:./output/ | ||
| Train.loader.batch_size_per_card:lite_train_lite_infer=2|whole_train_whole_infer=4 | ||
| Global.pretrained_model:pretrain_models/PPHGNet_small_ocr_det.pdparams | ||
| train_model_name:latest | ||
| train_infer_img_dir:./train_data/icdar2015/text_localization/ch4_test_images/ | ||
| null:null | ||
| ## | ||
| trainer:norm_train | ||
| norm_train:tools/train.py -c configs/det/ch_PP-OCRv4/ch_PP-OCRv4_det_teacher.yml -o Global.print_batch_step=1 Train.loader.shuffle=false Global.eval_batch_step=[4000,400] | ||
| pact_train:null | ||
| fpgm_train:null | ||
| distill_train:null | ||
| to_static_train:Global.to_static=true | ||
| null:null | ||
| ## | ||
| ===========================eval_params=========================== | ||
| eval:null | ||
| null:null | ||
| ## | ||
| ===========================infer_params=========================== | ||
| Global.save_inference_dir:./output/ | ||
| Global.checkpoints: | ||
| norm_export:tools/export_model.py -c configs/det/ch_PP-OCRv4/ch_PP-OCRv4_det_teacher.yml -o | ||
| quant_export:null | ||
| fpgm_export: | ||
| distill_export:null | ||
| export1:null | ||
| export2:null | ||
| inference_dir:Student | ||
| infer_model:./inference/ch_PP-OCRv3_det_infer/ | ||
| infer_export:null | ||
| infer_quant:False | ||
| inference:tools/infer/predict_det.py | ||
| --use_gpu:True|False | ||
| --enable_mkldnn:False | ||
| --cpu_threads:6 | ||
| --rec_batch_num:1 | ||
| --use_tensorrt:False | ||
| --precision:fp32 | ||
| --det_model_dir: | ||
| --image_dir:./inference/ch_det_data_50/all-sum-510/ | ||
| null:null | ||
| --benchmark:True | ||
| null:null | ||
| ===========================infer_benchmark_params========================== | ||
| random_infer_input:[{float32,[3,640,640]}];[{float32,[3,960,960]}] | ||
| ===========================train_benchmark_params========================== | ||
| batch_size:4 | ||
| fp_items:fp32|fp16 | ||
| epoch:2 | ||
| --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile | ||
| flags:FLAGS_eager_delete_tensor_gb=0.0;FLAGS_fraction_of_gpu_memory_to_use=0.98;FLAGS_conv_workspace_size_limit=4096 | ||
| ===========================to_static_train_benchmark_params=========================== | ||
| to_static_train:Global.to_static=true |
61 changes: 61 additions & 0 deletions
61
test_tipc/configs/ch_PP-OCRv4_server_rec/train_infer_python.txt
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,61 @@ | ||
| ===========================train_params=========================== | ||
| model_name:ch_PP-OCRv4_server_rec | ||
| python:python | ||
| gpu_list:0 | ||
| Global.use_gpu:True|True | ||
| Global.auto_cast:fp32 | ||
| Global.epoch_num:lite_train_lite_infer=3|whole_train_whole_infer=50 | ||
| Global.save_model_dir:./output/ | ||
| Train.loader.batch_size_per_card:lite_train_lite_infer=16|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 configs/rec/PP-OCRv4/ch_PP-OCRv4_rec_hgnet.yml -o Global.cal_metric_during_train=False Global.print_batch_step=1 Train.loader.shuffle=false Train.dataset.data_dir=./train_data/ic15_data Train.dataset.label_file_list=[./train_data/ic15_data/rec_gt_train.txt] Eval.dataset.data_dir=./train_data/ic15_data Eval.dataset.label_file_list=[./train_data/ic15_data/rec_gt_test.txt] Train.loader.num_workers=16 Eval.loader.num_workers=16 | ||
| pact_train:null | ||
| fpgm_train:null | ||
| distill_train:null | ||
| to_static_train:Global.to_static=true | ||
| null:null | ||
| ## | ||
| ===========================eval_params=========================== | ||
| eval:null | ||
| null:null | ||
| ## | ||
| ===========================infer_params=========================== | ||
| Global.save_inference_dir:./output/ | ||
| Global.checkpoints: | ||
| norm_export:tools/export_model.py -c configs/rec/PP-OCRv4/ch_PP-OCRv4_rec_hgnet.yml -o | ||
| quant_export: | ||
| fpgm_export: | ||
| distill_export:null | ||
| export1:null | ||
| export2:null | ||
| ## | ||
| infer_model:./inference/ch_PP-OCRv4_rec_infer | ||
| infer_export:null | ||
| infer_quant:False | ||
| inference:tools/infer/predict_rec.py --rec_image_shape="3,48,320" | ||
| --use_gpu:True|False | ||
| --enable_mkldnn:False | ||
| --cpu_threads:6 | ||
| --rec_batch_num:1 | ||
| --use_tensorrt:False | ||
| --precision:fp32 | ||
| --rec_model_dir: | ||
| --image_dir:./inference/rec_inference | ||
| null:null | ||
| --benchmark:True | ||
| null:null | ||
| ===========================infer_benchmark_params========================== | ||
| random_infer_input:[{float32,[3,48,320]}] | ||
| ===========================train_benchmark_params========================== | ||
| batch_size:128 | ||
| fp_items:fp32|fp16 | ||
| epoch:1 | ||
| --profiler_options:batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile | ||
| flags:FLAGS_eager_delete_tensor_gb=0.0;FLAGS_fraction_of_gpu_memory_to_use=0.98;FLAGS_conv_workspace_size_limit=4096 | ||
| ===========================to_static_train_benchmark_params=========================== | ||
| to_static_train:Global.to_static=true |
This file was deleted.
Oops, something went wrong.
Oops, something went wrong.