Fix the negative box size that could potentially trigger error.

PiperOrigin-RevId: 329957156

Fix the negative box size that could potentially trigger error.
PiperOrigin-RevId: 329957156
3cde342f · A. Unique TensorFlower · TF Object Detection Team · 575684c0 · 3cde342f
Commit 3cde342f authored Sep 03, 2020 by A. Unique TensorFlower Committed by TF Object Detection Team Sep 03, 2020
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

research/object_detection/meta_architectures/center_net_meta_arch.py ...ject_detection/meta_architectures/center_net_meta_arch.py +1 -0

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
@@ -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,
                           batch_dims=1)
+  height_width = tf.maximum(height_width, 0)
  offsets = tf.gather(offsets_flat, peak_spatial_indices, batch_dims=1)

  heights, widths = tf.unstack(height_width, axis=2)