Skip to content

GitLab

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

fixed dtype mismatch

parent 43081990
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/vision/beta/projects/panoptic_maskrcnn/modeling/layers/paste_masks.py
View file @ 9ab3bf77
......@@ -112,8 +112,8 @@ class PasteMasks(tf.keras.layers.Layer):
masks, boxes = inputs
y0, x0, y1, x1 = tf.split(boxes, 4, axis=1)
x_coords = tf.range(0, self._output_size[1], dtype=tf.float32)
y_coords = tf.range(0, self._output_size[0], dtype=tf.float32)
x_coords = tf.range(0, self._output_size[1], dtype=boxes.dtype)
y_coords = tf.range(0, self._output_size[0], dtype=boxes.dtype)
x_coords = (x_coords - x0) / (x1 - x0) * 2 - 1
y_coords = (y_coords - y0) / (y1 - y0) * 2 - 1
......
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