From 9f5d0d59efc326e3a9d4da368ae0c4b4ec8eb70f Mon Sep 17 00:00:00 2001 From: tink2123 Date: Wed, 27 Apr 2022 06:24:28 +0000 Subject: [PATCH] fix mkldnn for ppocrv3, and fix some typo --- ppocr/data/imaug/label_ops.py | 6 +++--- ppocr/utils/dict/ka_dict.txt | 2 +- ppocr/utils/dict/ta_dict.txt | 2 +- tools/infer/utility.py | 3 ++- 4 files changed, 7 insertions(+), 6 deletions(-) diff --git a/ppocr/data/imaug/label_ops.py b/ppocr/data/imaug/label_ops.py index 86366d7a..c9bc2e77 100644 --- a/ppocr/data/imaug/label_ops.py +++ b/ppocr/data/imaug/label_ops.py @@ -113,14 +113,14 @@ class BaseRecLabelEncode(object): dict_character = list(self.character_str) self.lower = True else: - self.character_str = "" + self.character_str = [] with open(character_dict_path, "rb") as fin: lines = fin.readlines() for line in lines: line = line.decode('utf-8').strip("\n").strip("\r\n") - self.character_str += line + self.character_str.append(line) if use_space_char: - self.character_str += " " + self.character_str.append(" ") dict_character = list(self.character_str) dict_character = self.add_special_char(dict_character) self.dict = {} diff --git a/ppocr/utils/dict/ka_dict.txt b/ppocr/utils/dict/ka_dict.txt index 33d605c4..d506b691 100644 --- a/ppocr/utils/dict/ka_dict.txt +++ b/ppocr/utils/dict/ka_dict.txt @@ -21,7 +21,7 @@ l 8 . j -p +p ಗ ು ಣ diff --git a/ppocr/utils/dict/ta_dict.txt b/ppocr/utils/dict/ta_dict.txt index d1bae501..19d81892 100644 --- a/ppocr/utils/dict/ta_dict.txt +++ b/ppocr/utils/dict/ta_dict.txt @@ -22,7 +22,7 @@ l 8 . j -p +p ப ூ த diff --git a/tools/infer/utility.py b/tools/infer/utility.py index 762db868..a93f364b 100644 --- a/tools/infer/utility.py +++ b/tools/infer/utility.py @@ -301,7 +301,8 @@ def create_predictor(args, mode, logger): # enable memory optim config.enable_memory_optim() config.disable_glog_info() - + config.delete_pass("reshape_transpose_matmul_v2_mkldnn_fuse_pass") + config.delete_pass("reshape_transpose_matmul_mkldnn_fuse_pass") config.delete_pass("conv_transpose_eltwiseadd_bn_fuse_pass") if mode == 'table': config.delete_pass("fc_fuse_pass") # not supported for table -- GitLab