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
4eba6c0d
Commit
4eba6c0d
authored
Nov 09, 2020
by
WenmuZhou
Browse files
修复一些导致不可用的bug
parent
49958dca
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
13 deletions
+5
-13
tools/eval.py
tools/eval.py
+5
-13
No files found.
tools/eval.py
View file @
4eba6c0d
...
@@ -23,12 +23,8 @@ __dir__ = os.path.dirname(os.path.abspath(__file__))
...
@@ -23,12 +23,8 @@ __dir__ = os.path.dirname(os.path.abspath(__file__))
sys
.
path
.
append
(
__dir__
)
sys
.
path
.
append
(
__dir__
)
sys
.
path
.
append
(
os
.
path
.
abspath
(
os
.
path
.
join
(
__dir__
,
'..'
)))
sys
.
path
.
append
(
os
.
path
.
abspath
(
os
.
path
.
join
(
__dir__
,
'..'
)))
import
paddle
# paddle.manual_seed(2)
from
ppocr.utils.logging
import
get_logger
from
ppocr.data
import
build_dataloader
from
ppocr.data
import
build_dataloader
from
ppocr.modeling
import
build_model
from
ppocr.modeling
.architectures
import
build_model
from
ppocr.postprocess
import
build_post_process
from
ppocr.postprocess
import
build_post_process
from
ppocr.metrics
import
build_metric
from
ppocr.metrics
import
build_metric
from
ppocr.utils.save_load
import
init_model
from
ppocr.utils.save_load
import
init_model
...
@@ -39,8 +35,7 @@ import tools.program as program
...
@@ -39,8 +35,7 @@ import tools.program as program
def
main
():
def
main
():
global_config
=
config
[
'Global'
]
global_config
=
config
[
'Global'
]
# build dataloader
# build dataloader
eval_loader
,
_
=
build_dataloader
(
config
[
'EVAL'
],
device
,
False
,
valid_dataloader
=
build_dataloader
(
config
,
'Eval'
,
device
,
logger
)
global_config
)
# build post process
# build post process
post_process_class
=
build_post_process
(
config
[
'PostProcess'
],
post_process_class
=
build_post_process
(
config
[
'PostProcess'
],
...
@@ -63,16 +58,13 @@ def main():
...
@@ -63,16 +58,13 @@ def main():
eval_class
=
build_metric
(
config
[
'Metric'
])
eval_class
=
build_metric
(
config
[
'Metric'
])
# start eval
# start eval
metirc
=
program
.
eval
(
model
,
eval_loader
,
post_process_class
,
eval_class
)
metirc
=
program
.
eval
(
model
,
valid_dataloader
,
post_process_class
,
eval_class
)
logger
.
info
(
'metric eval ***************'
)
logger
.
info
(
'metric eval ***************'
)
for
k
,
v
in
metirc
.
items
():
for
k
,
v
in
metirc
.
items
():
logger
.
info
(
'{}:{}'
.
format
(
k
,
v
))
logger
.
info
(
'{}:{}'
.
format
(
k
,
v
))
if
__name__
==
'__main__'
:
if
__name__
==
'__main__'
:
device
,
config
=
program
.
preprocess
()
config
,
device
,
logger
,
vdl_writer
=
program
.
preprocess
()
paddle
.
disable_static
(
device
)
logger
=
get_logger
()
print_dict
(
config
,
logger
)
main
()
main
()
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