Skip to content

GitLab

"git@developer.sourcefind.cn:modelzoo/resnet50_tensorflow.git" did not exist on "f3600cd1f755090168ea44d85d47161fafb37bc8"
Unverified Commit 401b105b authored by srihari-humbarwadi's avatar srihari-humbarwadi
Browse files

fixed dtype mismatch in `_paste_mask`

parent ed21b802
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
official/vision/beta/projects/panoptic_maskrcnn/modeling/layers/panoptic_segmentation_generator.py
View file @ 401b105b
...@@ -92,7 +92,8 @@ class PanopticSegmentationGenerator(tf.keras.layers.Layer): ...@@ -92,7 +92,8 @@ class PanopticSegmentationGenerator(tf.keras.layers.Layer):
super(PanopticSegmentationGenerator, self).__init__(**kwargs) super(PanopticSegmentationGenerator, self).__init__(**kwargs)
def _paste_mask(self, box, mask): def _paste_mask(self, box, mask):
pasted_mask = tf.ones(self._output_size + [1]) * self._void_class_label pasted_mask = tf.ones(
self._output_size + [1], dtype=mask.dtype) * self._void_class_label
ymin = box[0] ymin = box[0]
xmin = box[1] xmin = box[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