Commit 30cc8e74 authored by Vivek Rathod's avatar Vivek Rathod Committed by TF Object Detection Team
Browse files

Make label_id_offset an argument so it can be turned off when...

Make label_id_offset an argument so it can be turned off when eval_util.result_dict_for_batched_example is applied on the outputs of an exported SavedModel. Exported saved models already apply label offset.

PiperOrigin-RevId: 341176846
parent 90d1a0bb
...@@ -770,7 +770,8 @@ def result_dict_for_batched_example(images, ...@@ -770,7 +770,8 @@ def result_dict_for_batched_example(images,
scale_to_absolute=False, scale_to_absolute=False,
original_image_spatial_shapes=None, original_image_spatial_shapes=None,
true_image_shapes=None, true_image_shapes=None,
max_gt_boxes=None): max_gt_boxes=None,
label_id_offset=1):
"""Merges all detection and groundtruth information for a single example. """Merges all detection and groundtruth information for a single example.
Note that evaluation tools require classes that are 1-indexed, and so this Note that evaluation tools require classes that are 1-indexed, and so this
...@@ -825,6 +826,7 @@ def result_dict_for_batched_example(images, ...@@ -825,6 +826,7 @@ def result_dict_for_batched_example(images,
containing the size of the unpadded original_image. containing the size of the unpadded original_image.
max_gt_boxes: [batch_size] tensor representing the maximum number of max_gt_boxes: [batch_size] tensor representing the maximum number of
groundtruth boxes to pad. groundtruth boxes to pad.
label_id_offset: offset for class ids.
Returns: Returns:
A dictionary with: A dictionary with:
...@@ -879,8 +881,6 @@ def result_dict_for_batched_example(images, ...@@ -879,8 +881,6 @@ def result_dict_for_batched_example(images,
ValueError: if true_image_shapes is not 2D int32 tensor of shape ValueError: if true_image_shapes is not 2D int32 tensor of shape
[3]. [3].
""" """
label_id_offset = 1 # Applying label id offset (b/63711816)
input_data_fields = fields.InputDataFields input_data_fields = fields.InputDataFields
if original_image_spatial_shapes is None: if original_image_spatial_shapes is None:
original_image_spatial_shapes = tf.tile( original_image_spatial_shapes = tf.tile(
......
Markdown is supported
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