From 6e03b3de14964c3a3b095fb04d2f04b7e7bf8014 Mon Sep 17 00:00:00 2001
From: WenmuZhou <zjwenmu@gmail.com>
Date: Thu, 10 Jun 2021 17:29:23 +0800
Subject: [PATCH] del unused params

---
 ppocr/postprocess/rec_postprocess.py   | 18 ------------------
 ppstructure/table/predict_structure.py |  3 ---
 ppstructure/utility.py                 |  3 ---
 3 files changed, 24 deletions(-)

diff --git a/ppocr/postprocess/rec_postprocess.py b/ppocr/postprocess/rec_postprocess.py
index 164dec55..8426bcf2 100644
--- a/ppocr/postprocess/rec_postprocess.py
+++ b/ppocr/postprocess/rec_postprocess.py
@@ -325,14 +325,8 @@ class TableLabelDecode(object):
     """  """
 
     def __init__(self,
-                 max_text_length,
-                 max_elem_length,
-                 max_cell_num,
                  character_dict_path,
                  **kwargs):
-        self.max_text_length = max_text_length
-        self.max_elem_length = max_elem_length
-        self.max_cell_num = max_cell_num
         list_character, list_elem = self.load_char_elem_dict(character_dict_path)
         list_character = self.add_special_char(list_character)
         list_elem = self.add_special_char(list_elem)
@@ -369,18 +363,6 @@ class TableLabelDecode(object):
         list_character = [self.beg_str] + list_character + [self.end_str]
         return list_character
 
-    def get_sp_tokens(self):
-        char_beg_idx = self.get_beg_end_flag_idx('beg', 'char')
-        char_end_idx = self.get_beg_end_flag_idx('end', 'char')
-        elem_beg_idx = self.get_beg_end_flag_idx('beg', 'elem')
-        elem_end_idx = self.get_beg_end_flag_idx('end', 'elem')
-        elem_char_idx1 = self.dict_elem['<td>']
-        elem_char_idx2 = self.dict_elem['<td']
-        sp_tokens = np.array([char_beg_idx, char_end_idx, elem_beg_idx,
-                              elem_end_idx, elem_char_idx1, elem_char_idx2, self.max_text_length,
-                              self.max_elem_length, self.max_cell_num])
-        return sp_tokens
-
     def __call__(self, preds):
         structure_probs = preds['structure_probs']
         loc_preds = preds['loc_preds']
diff --git a/ppstructure/table/predict_structure.py b/ppstructure/table/predict_structure.py
index b576682c..d8a3b3a8 100755
--- a/ppstructure/table/predict_structure.py
+++ b/ppstructure/table/predict_structure.py
@@ -63,9 +63,6 @@ class TableStructurer(object):
             'name': 'TableLabelDecode',
             "character_type": args.structure_char_type,
             "character_dict_path": args.structure_char_dict_path,
-            "max_text_length": args.structure_max_text_length,
-            "max_elem_length": args.structure_max_elem_length,
-            "max_cell_num": args.structure_max_cell_num
         }
 
         self.preprocess_op = create_operators(pre_process_list)
diff --git a/ppstructure/utility.py b/ppstructure/utility.py
index 134d2d7b..0dd220af 100644
--- a/ppstructure/utility.py
+++ b/ppstructure/utility.py
@@ -24,9 +24,6 @@ def init_args():
     parser.add_argument("--output", type=str, default='./output/table')
     # params for table structure
     parser.add_argument("--structure_max_len", type=int, default=488)
-    parser.add_argument("--structure_max_text_length", type=int, default=100)
-    parser.add_argument("--structure_max_elem_length", type=int, default=800)
-    parser.add_argument("--structure_max_cell_num", type=int, default=500)
     parser.add_argument("--structure_model_dir", type=str)
     parser.add_argument("--structure_char_type", type=str, default='en')
     parser.add_argument("--structure_char_dict_path", type=str, default="../ppocr/utils/dict/table_structure_dict.txt")
-- 
GitLab