diff --git a/.gitignore b/.gitignore index caf886a2b581b5976ea391aca5d7a56041bdbaa8..70f136ccff632b77df358643e5815eb0bf6b0395 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 3e1d3009ce42981a191afbbc9375b9898a9c3004..271ae8fd327acb514ba5eafd60205a0c8d120364 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 a49f332aae0eeb1b7a9d728d37095e5d88c3e6f5..83db9461d931c5271a188b88c4fc1600bd248917 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