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
ModelZoo
ResNet50_tensorflow
Commits
0a8036ce
Commit
0a8036ce
authored
Mar 02, 2021
by
Ronny Votel
Committed by
TF Object Detection Team
Mar 02, 2021
Browse files
Internal change.
PiperOrigin-RevId: 360482921
parent
8c169ea4
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
11 additions
and
4 deletions
+11
-4
research/object_detection/model_lib_v2.py
research/object_detection/model_lib_v2.py
+11
-4
No files found.
research/object_detection/model_lib_v2.py
View file @
0a8036ce
...
...
@@ -795,7 +795,8 @@ def eager_eval_loop(
eval_dataset
,
use_tpu
=
False
,
postprocess_on_cpu
=
False
,
global_step
=
None
):
global_step
=
None
,
):
"""Evaluate the model eagerly on the evaluation dataset.
This method will compute the evaluation metrics specified in the configs on
...
...
@@ -968,11 +969,10 @@ def eager_eval_loop(
eval_metrics
[
loss_key
]
=
tf
.
reduce_mean
(
loss_metrics
[
loss_key
])
eval_metrics
=
{
str
(
k
):
v
for
k
,
v
in
eval_metrics
.
items
()}
tf
.
logging
.
info
(
'Eval metrics at step %d'
,
global_step
)
tf
.
logging
.
info
(
'Eval metrics at step %d'
,
global_step
.
numpy
()
)
for
k
in
eval_metrics
:
tf
.
compat
.
v2
.
summary
.
scalar
(
k
,
eval_metrics
[
k
],
step
=
global_step
)
tf
.
logging
.
info
(
'
\t
+ %s: %f'
,
k
,
eval_metrics
[
k
])
return
eval_metrics
...
...
@@ -1026,6 +1026,8 @@ def eval_continuously(
"""
get_configs_from_pipeline_file
=
MODEL_BUILD_UTIL_MAP
[
'get_configs_from_pipeline_file'
]
create_pipeline_proto_from_configs
=
MODEL_BUILD_UTIL_MAP
[
'create_pipeline_proto_from_configs'
]
merge_external_params_with_configs
=
MODEL_BUILD_UTIL_MAP
[
'merge_external_params_with_configs'
]
...
...
@@ -1043,6 +1045,10 @@ def eval_continuously(
'Forced number of epochs for all eval validations to be 1.'
)
configs
=
merge_external_params_with_configs
(
configs
,
None
,
kwargs_dict
=
kwargs
)
if
model_dir
:
pipeline_config_final
=
create_pipeline_proto_from_configs
(
configs
)
config_util
.
save_pipeline_config
(
pipeline_config_final
,
model_dir
)
model_config
=
configs
[
'model'
]
train_input_config
=
configs
[
'train_input_config'
]
eval_config
=
configs
[
'eval_config'
]
...
...
@@ -1109,4 +1115,5 @@ def eval_continuously(
eval_input
,
use_tpu
=
use_tpu
,
postprocess_on_cpu
=
postprocess_on_cpu
,
global_step
=
global_step
)
global_step
=
global_step
,
)
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