Skip to content

GitLab

Commit cf3bb8cf authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 403132893
parent 63efa179
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
official/vision/beta/modeling/layers/detection_generator.py
View file @ cf3bb8cf
...@@ -391,6 +391,7 @@ def _generate_detections_batched(boxes: tf.Tensor, scores: tf.Tensor, ...@@ -391,6 +391,7 @@ def _generate_detections_batched(boxes: tf.Tensor, scores: tf.Tensor,
score_threshold=pre_nms_score_threshold, score_threshold=pre_nms_score_threshold,
pad_per_class=False, pad_per_class=False,
clip_boxes=False)) clip_boxes=False))
nmsed_classes = tf.cast(nmsed_classes, tf.int32)
return nmsed_boxes, nmsed_scores, nmsed_classes, valid_detections return nmsed_boxes, nmsed_scores, nmsed_classes, valid_detections
......
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