attention ocr

zhang0jhon · Nov 7, 2019 · ba46051 · ba46051
1 parent a2eb8da
commit ba46051
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/parse_dict.py b/parse_dict.py
@@ -27,9 +27,11 @@ def get_dict(path=os.path.join(currentdir, 'label_dict/icdar_labels.txt'), add_s
  idx, label = int(m.group(1)), m.group(2)
  label_dict[idx] = label 
  if add_space:
- label_dict[idx+1] = ' ' 
+ idx = idx + 1
+ label_dict[idx] = ' ' 
  if add_eos:
- label_dict[idx+2] = 'EOS' 
+ idx = idx + 1
+ label_dict[idx] = 'EOS' 
  return label_dict