Skip to content

GitLab

Commit 93da8e77 authored by Vivek Rathod's avatar Vivek Rathod Committed by TF Object Detection Team
Browse files

Add explicit static shapes to prevent export errors. 

PiperOrigin-RevId: 362204802
parent 3570bd34
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
research/object_detection/meta_architectures/center_net_meta_arch.py
View file @ 93da8e77
......@@ -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)
......
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