Skip to content
Snippets Groups Projects
Commit 4784fc62 authored by Daniel Ecer's avatar Daniel Ecer
Browse files

removed loss from summary filename

parent 1eb2cecd
No related branches found
No related tags found
No related merge requests found
...@@ -201,7 +201,6 @@ class Evaluator(object): ...@@ -201,7 +201,6 @@ class Evaluator(object):
def _save_prediction_summary_image(self, eval_index, results): def _save_prediction_summary_image(self, eval_index, results):
logger = get_logger() logger = get_logger()
global_step = results['global_step'] global_step = results['global_step']
metric_values = results['metric_values']
for batch_index, input_uri in enumerate(results['input_uri']): for batch_index, input_uri in enumerate(results['input_uri']):
pred_image = results['input_image'][batch_index] pred_image = results['input_image'][batch_index]
pred_annotation = results['annotation_image'][batch_index] pred_annotation = results['annotation_image'][batch_index]
...@@ -214,8 +213,8 @@ class Evaluator(object): ...@@ -214,8 +213,8 @@ class Evaluator(object):
pred_np pred_np
) )
result_file = os.path.join( result_file = os.path.join(
self.results_dir, 'result_{}_{}_{}_summary_{}.png'.format( self.results_dir, 'result_{}_{}_{}_summary.png'.format(
global_step, eval_index, batch_index, metric_values[0] global_step, eval_index, batch_index
) )
) )
logging.info('result_file: %s', result_file) logging.info('result_file: %s', result_file)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment