From c53a6e5137c231e4a738bcf99b2e805e9ad4cc97 Mon Sep 17 00:00:00 2001 From: smilelite Date: Mon, 11 Jul 2022 23:28:51 +0800 Subject: [PATCH] modified config --- .gitignore | 2 ++ configs/rec/rec_r31_robustscanner.yml | 2 +- .../rec_r31_robustscanner/rec_r31_robustscanner.yml | 8 ++++---- 3 files changed, 7 insertions(+), 5 deletions(-) diff --git a/.gitignore b/.gitignore index caf886a2..70f136cc 100644 --- a/.gitignore +++ b/.gitignore @@ -10,6 +10,8 @@ __pycache__/ inference/ inference_results/ output/ +train_data +log *.DS_Store *.vs diff --git a/configs/rec/rec_r31_robustscanner.yml b/configs/rec/rec_r31_robustscanner.yml index 3e1d3009..271ae8fd 100644 --- a/configs/rec/rec_r31_robustscanner.yml +++ b/configs/rec/rec_r31_robustscanner.yml @@ -96,7 +96,7 @@ Eval: - SARLabelEncode: # Class handling label - RobustScannerRecResizeImg: image_shape: [3, 48, 48, 160] - max_seq_len: *max_text_length + max_text_length: *max_text_length width_downsample_ratio: 0.25 - KeepKeys: keep_keys: ['image', 'label', 'valid_ratio', 'word_positons'] # dataloader will return list in this order diff --git a/test_tipc/configs/rec_r31_robustscanner/rec_r31_robustscanner.yml b/test_tipc/configs/rec_r31_robustscanner/rec_r31_robustscanner.yml index a49f332a..83db9461 100644 --- a/test_tipc/configs/rec_r31_robustscanner/rec_r31_robustscanner.yml +++ b/test_tipc/configs/rec_r31_robustscanner/rec_r31_robustscanner.yml @@ -12,7 +12,7 @@ Global: checkpoints: save_inference_dir: use_visualdl: False - infer_img: + infer_img: ./inference/rec_inference # for data or label process character_dict_path: ppocr/utils/dict90.txt max_text_length: &max_text_length 40 @@ -49,7 +49,7 @@ Architecture: mask: True padding_idx: 92 encode_value: False - max_seq_len: 40 + max_text_length: *max_text_length Loss: name: SARLoss @@ -75,7 +75,7 @@ Train: - RobustScannerRecResizeImg: image_shape: [3, 48, 48, 160] # h:48 w:[48,160] width_downsample_ratio: 0.25 - max_seq_len: *max_text_length + max_text_length: *max_text_length - KeepKeys: keep_keys: ['image', 'label', 'valid_ratio', 'word_positons'] # dataloader will return list in this order loader: @@ -97,7 +97,7 @@ Eval: - SARLabelEncode: # Class handling label - RobustScannerRecResizeImg: image_shape: [3, 48, 48, 160] - max_seq_len: *max_text_length + max_text_length: *max_text_length width_downsample_ratio: 0.25 - KeepKeys: keep_keys: ['image', 'label', 'valid_ratio', 'word_positons'] # dataloader will return list in this order -- GitLab