Skip to content

GitLab

Commit 3cde342f authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TF Object Detection Team
Browse files

Fix the negative box size that could potentially trigger error. 

PiperOrigin-RevId: 329957156
parent 575684c0
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
research/object_detection/meta_architectures/center_net_meta_arch.py
View file @ 3cde342f
...@@ -311,6 +311,7 @@ def prediction_tensors_to_boxes(detection_scores, y_indices, x_indices, ...@@ -311,6 +311,7 @@ def prediction_tensors_to_boxes(detection_scores, y_indices, x_indices,
height_width = tf.gather(height_width_flat, peak_spatial_indices, height_width = tf.gather(height_width_flat, peak_spatial_indices,
batch_dims=1) batch_dims=1)
height_width = tf.maximum(height_width, 0)
offsets = tf.gather(offsets_flat, peak_spatial_indices, batch_dims=1) offsets = tf.gather(offsets_flat, peak_spatial_indices, batch_dims=1)
heights, widths = tf.unstack(height_width, axis=2) heights, widths = tf.unstack(height_width, axis=2)
......
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