Skip to content

GitLab

You need to sign in or sign up before continuing.
Commit 3cc082ea authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 403132893
parent 8513f442
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
official/vision/beta/modeling/layers/detection_generator.py
View file @ 3cc082ea
...@@ -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