提交 5983e1af 编写于 作者: W WenmuZhou

Remove redundant functions

上级 835bfa45
...@@ -123,7 +123,7 @@ class BaseRecLabelEncode(object): ...@@ -123,7 +123,7 @@ class BaseRecLabelEncode(object):
[sum(text_lengths)] = [text_index_0 + text_index_1 + ... + text_index_(n - 1)] [sum(text_lengths)] = [text_index_0 + text_index_1 + ... + text_index_(n - 1)]
length: length of each text. [batch_size] length: length of each text. [batch_size]
""" """
if len(text) > self.max_text_len: if len(text) == 0 or len(text) > self.max_text_len:
return None return None
if self.character_type == "en": if self.character_type == "en":
text = text.lower() text = text.lower()
...@@ -138,9 +138,6 @@ class BaseRecLabelEncode(object): ...@@ -138,9 +138,6 @@ class BaseRecLabelEncode(object):
return None return None
return text_list return text_list
def get_ignored_tokens(self):
return [0] # for ctc blank
class CTCLabelEncode(BaseRecLabelEncode): class CTCLabelEncode(BaseRecLabelEncode):
""" Convert between text-label and text-index """ """ Convert between text-label and text-index """
...@@ -193,11 +190,6 @@ class AttnLabelEncode(BaseRecLabelEncode): ...@@ -193,11 +190,6 @@ class AttnLabelEncode(BaseRecLabelEncode):
text = self.encode(text) text = self.encode(text)
return text return text
def get_ignored_tokens(self):
beg_idx = self.get_beg_end_flag_idx("beg")
end_idx = self.get_beg_end_flag_idx("end")
return [beg_idx, end_idx]
def get_beg_end_flag_idx(self, beg_or_end): def get_beg_end_flag_idx(self, beg_or_end):
if beg_or_end == "beg": if beg_or_end == "beg":
idx = np.array(self.dict[self.beg_str]) idx = np.array(self.dict[self.beg_str])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册