Add explicit static shapes to prevent export errors.

PiperOrigin-RevId: 362204802

Add explicit static shapes to prevent export errors.
PiperOrigin-RevId: 362204802
93da8e77 · Vivek Rathod · TF Object Detection Team · 3570bd34 · 93da8e77
Commit 93da8e77 authored Mar 10, 2021 by Vivek Rathod Committed by TF Object Detection Team Mar 10, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

research/object_detection/meta_architectures/center_net_meta_arch.py ...ject_detection/meta_architectures/center_net_meta_arch.py +2 -2

No files found.
--- a/research/object_detection/meta_architectures/center_net_meta_arch.py
+++ b/research/object_detection/meta_architectures/center_net_meta_arch.py
@@ -375,10 +375,10 @@ def prediction_tensors_to_boxes(detection_scores, y_indices, x_indices,
      tf.reshape(x_indices, [-1])
  ], axis=1)
  new_height_width = tf.gather_nd(height_width_predictions, combined_indices)
-  new_height_width = tf.reshape(new_height_width, [batch_size, num_boxes, -1])
+  new_height_width = tf.reshape(new_height_width, [batch_size, num_boxes, 2])

  new_offsets = tf.gather_nd(offset_predictions, combined_indices)
-  offsets = tf.reshape(new_offsets, [batch_size, num_boxes, -1])
+  offsets = tf.reshape(new_offsets, [batch_size, num_boxes, 2])

  y_indices = _to_float32(y_indices)
  x_indices = _to_float32(x_indices)