Skip to content

Commit

Permalink
No public description
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 631335119
  • Loading branch information
tensorflower-gardener committed May 7, 2024
1 parent ae79c47 commit a0d0925
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions official/vision/modeling/layers/detection_generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -910,7 +910,7 @@ def dummy_post_processing(input_boxes, input_scores, input_anchors):
dummy_num_detections = tf.constant(
0.0, dtype=tf.float32, name='num_detections')
return boxes, dummy_classes, scores, dummy_num_detections
return dummy_post_processing(boxes, scores, anchors)[::-1]
return dummy_post_processing(boxes, scores, anchors)


@tf_keras.utils.register_keras_serializable(package='Vision')
Expand Down Expand Up @@ -1481,10 +1481,10 @@ def __call__(
self._config_dict['box_coder_weights'],
)
return {
'num_detections': num_detections,
'detection_boxes': boxes,
'detection_classes': classes,
'detection_scores': scores,
'num_detections': num_detections,
}

if self._config_dict['nms_version'] != 'v3':
Expand Down

0 comments on commit a0d0925

Please sign in to comment.