Internal change

PiperOrigin-RevId: 410385368

Internal change
PiperOrigin-RevId: 410385368
a76a6b94 · A. Unique TensorFlower · e0a16ee8 · a76a6b94
Commit a76a6b94 authored Nov 16, 2021 by A. Unique TensorFlower
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/vision/beta/modeling/layers/detection_generator.py official/vision/beta/modeling/layers/detection_generator.py +1 -1

No files found.
--- a/official/vision/beta/modeling/layers/detection_generator.py
+++ b/official/vision/beta/modeling/layers/detection_generator.py
@@ -368,7 +368,7 @@ def _generate_detections_v2(boxes: tf.Tensor,
  nmsed_boxes = tf.gather(nmsed_boxes, indices, batch_dims=1, axis=1)
  nmsed_classes = tf.gather(nmsed_classes, indices, batch_dims=1)
  valid_detections = tf.reduce_sum(
-      input_tensor=tf.cast(tf.greater(nmsed_scores, -1), tf.int32), axis=1)
+      input_tensor=tf.cast(tf.greater(nmsed_scores, 0.0), tf.int32), axis=1)
  return nmsed_boxes, nmsed_scores, nmsed_classes, valid_detections