Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
wangsen
paddle_dbnet
Commits
9f5d0d59
Commit
9f5d0d59
authored
Apr 27, 2022
by
tink2123
Browse files
fix mkldnn for ppocrv3, and fix some typo
parent
58a408ab
Changes
4
Show whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
7 additions
and
6 deletions
+7
-6
ppocr/data/imaug/label_ops.py
ppocr/data/imaug/label_ops.py
+3
-3
ppocr/utils/dict/ka_dict.txt
ppocr/utils/dict/ka_dict.txt
+1
-1
ppocr/utils/dict/ta_dict.txt
ppocr/utils/dict/ta_dict.txt
+1
-1
tools/infer/utility.py
tools/infer/utility.py
+2
-1
No files found.
ppocr/data/imaug/label_ops.py
View file @
9f5d0d59
...
@@ -113,14 +113,14 @@ class BaseRecLabelEncode(object):
...
@@ -113,14 +113,14 @@ class BaseRecLabelEncode(object):
dict_character
=
list
(
self
.
character_str
)
dict_character
=
list
(
self
.
character_str
)
self
.
lower
=
True
self
.
lower
=
True
else
:
else
:
self
.
character_str
=
""
self
.
character_str
=
[]
with
open
(
character_dict_path
,
"rb"
)
as
fin
:
with
open
(
character_dict_path
,
"rb"
)
as
fin
:
lines
=
fin
.
readlines
()
lines
=
fin
.
readlines
()
for
line
in
lines
:
for
line
in
lines
:
line
=
line
.
decode
(
'utf-8'
).
strip
(
"
\n
"
).
strip
(
"
\r\n
"
)
line
=
line
.
decode
(
'utf-8'
).
strip
(
"
\n
"
).
strip
(
"
\r\n
"
)
self
.
character_str
+=
line
self
.
character_str
.
append
(
line
)
if
use_space_char
:
if
use_space_char
:
self
.
character_str
+=
" "
self
.
character_str
.
append
(
" "
)
dict_character
=
list
(
self
.
character_str
)
dict_character
=
list
(
self
.
character_str
)
dict_character
=
self
.
add_special_char
(
dict_character
)
dict_character
=
self
.
add_special_char
(
dict_character
)
self
.
dict
=
{}
self
.
dict
=
{}
...
...
ppocr/utils/dict/ka_dict.txt
View file @
9f5d0d59
ppocr/utils/dict/ta_dict.txt
View file @
9f5d0d59
tools/infer/utility.py
View file @
9f5d0d59
...
@@ -301,7 +301,8 @@ def create_predictor(args, mode, logger):
...
@@ -301,7 +301,8 @@ def create_predictor(args, mode, logger):
# enable memory optim
# enable memory optim
config
.
enable_memory_optim
()
config
.
enable_memory_optim
()
config
.
disable_glog_info
()
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"
)
config
.
delete_pass
(
"conv_transpose_eltwiseadd_bn_fuse_pass"
)
if
mode
==
'table'
:
if
mode
==
'table'
:
config
.
delete_pass
(
"fc_fuse_pass"
)
# not supported for table
config
.
delete_pass
(
"fc_fuse_pass"
)
# not supported for table
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment