diff --git a/configs/rec/rec_r31_sar.yml b/configs/rec/rec_r31_sar.yml index 73d56f375e81e70166fcf499f372afb2dd2fb5c2..7adb7136c50bbdfa764e33126bfc15f52510a9f1 100644 --- a/configs/rec/rec_r31_sar.yml +++ b/configs/rec/rec_r31_sar.yml @@ -19,6 +19,7 @@ Global: max_text_length: 30 infer_mode: False use_space_char: False + rm_symbol: True save_res_path: ./output/rec/predicts_sar.txt Optimizer: @@ -47,7 +48,6 @@ Loss: PostProcess: name: SARLabelDecode - rm_symbol: True Metric: name: RecMetric diff --git a/ppocr/postprocess/rec_postprocess.py b/ppocr/postprocess/rec_postprocess.py index 2043928142b6f59d60d5256bb890a88e062f8fd6..f2c3f289b87a883f94314e2baf30fddc3d9544a1 100644 --- a/ppocr/postprocess/rec_postprocess.py +++ b/ppocr/postprocess/rec_postprocess.py @@ -526,10 +526,10 @@ class SARLabelDecode(BaseRecLabelDecode): character_dict_path=None, character_type='ch', use_space_char=False, - rm_symbol=True, **kwargs): super(SARLabelDecode, self).__init__(character_dict_path, character_type, use_space_char) + self.rm_symbol = kwargs.get('rm_symbol', True) def add_special_char(self, dict_character): beg_end_str = ""