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
552d4274
Commit
552d4274
authored
May 25, 2020
by
tink2123
Browse files
fix eval_rec
parent
dd0112f5
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
4 deletions
+4
-4
tools/eval_utils/eval_rec_utils.py
tools/eval_utils/eval_rec_utils.py
+4
-4
No files found.
tools/eval_utils/eval_rec_utils.py
View file @
552d4274
...
@@ -48,7 +48,7 @@ def eval_rec_run(exe, config, eval_info_dict, mode):
...
@@ -48,7 +48,7 @@ def eval_rec_run(exe, config, eval_info_dict, mode):
total_sample_num
=
0
total_sample_num
=
0
total_acc_num
=
0
total_acc_num
=
0
total_batch_num
=
0
total_batch_num
=
0
if
mode
==
"
eval
"
:
if
mode
==
"
test
"
:
is_remove_duplicate
=
False
is_remove_duplicate
=
False
else
:
else
:
is_remove_duplicate
=
True
is_remove_duplicate
=
True
...
@@ -91,11 +91,11 @@ def test_rec_benchmark(exe, config, eval_info_dict):
...
@@ -91,11 +91,11 @@ def test_rec_benchmark(exe, config, eval_info_dict):
total_correct_number
=
0
total_correct_number
=
0
eval_data_acc_info
=
{}
eval_data_acc_info
=
{}
for
eval_data
in
eval_data_list
:
for
eval_data
in
eval_data_list
:
config
[
'
Test
Reader'
][
'lmdb_sets_dir'
]
=
\
config
[
'
Eval
Reader'
][
'lmdb_sets_dir'
]
=
\
eval_data_dir
+
"/"
+
eval_data
eval_data_dir
+
"/"
+
eval_data
eval_reader
=
reader_main
(
config
=
config
,
mode
=
"
test
"
)
eval_reader
=
reader_main
(
config
=
config
,
mode
=
"
eval
"
)
eval_info_dict
[
'reader'
]
=
eval_reader
eval_info_dict
[
'reader'
]
=
eval_reader
metrics
=
eval_rec_run
(
exe
,
config
,
eval_info_dict
,
"
test
"
)
metrics
=
eval_rec_run
(
exe
,
config
,
eval_info_dict
,
"
eval
"
)
total_evaluation_data_number
+=
metrics
[
'total_sample_num'
]
total_evaluation_data_number
+=
metrics
[
'total_sample_num'
]
total_correct_number
+=
metrics
[
'total_acc_num'
]
total_correct_number
+=
metrics
[
'total_acc_num'
]
eval_data_acc_info
[
eval_data
]
=
metrics
eval_data_acc_info
[
eval_data
]
=
metrics
...
...
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