"vscode:/vscode.git/clone" did not exist on "dcd76d49cde67518b155f800aeedadce2a00c64a"
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
......@@ -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