fix mkldnn for ppocrv3, and fix some typo
parent
58a408abe0
commit
9f5d0d59ef
ppocr
data/imaug
utils/dict
tools/infer
|
@ -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 = {}
|
||||
|
|
|
@ -21,7 +21,7 @@ l
|
|||
8
|
||||
.
|
||||
j
|
||||
p
|
||||
p
|
||||
ಗ
|
||||
ು
|
||||
ಣ
|
||||
|
|
|
@ -22,7 +22,7 @@ l
|
|||
8
|
||||
.
|
||||
j
|
||||
p
|
||||
p
|
||||
ப
|
||||
ூ
|
||||
த
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue