Skip to content

GitLab

Unverified Commit d8eed145 authored by srihari-humbarwadi's avatar srihari-humbarwadi
Browse files

fixed incorrect axis

parent 52c7f869
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/vision/beta/projects/panoptic_maskrcnn/modeling/layers/panoptic_segmentation_generator.py
View file @ d8eed145
...@@ -244,7 +244,7 @@ class PanopticSegmentationGenerator(tf.keras.layers.Layer): ...@@ -244,7 +244,7 @@ class PanopticSegmentationGenerator(tf.keras.layers.Layer):
image_shape = tf.cast(image_shape, dtype=batched_boxes.dtype) image_shape = tf.cast(image_shape, dtype=batched_boxes.dtype)
scale = tf.convert_to_tensor( scale = tf.convert_to_tensor(
[self._output_size], dtype=batched_boxes.dtype) / image_shape [self._output_size], dtype=batched_boxes.dtype) / image_shape
scale = tf.tile(tf.expand_dims(scale, axis=0), multiples=[1, 1, 2]) scale = tf.tile(tf.expand_dims(scale, axis=1), multiples=[1, 1, 2])
batched_boxes = batched_boxes * scale batched_boxes = batched_boxes * scale
panoptic_masks = tf.map_fn( panoptic_masks = tf.map_fn(
......
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