Skip to content

GitLab

"...git@developer.sourcefind.cn:OpenDAS/torch-cluster.git" did not exist on "e95d789bfda0947dbfb1fe8c26de6c9eddb2db71"
Unverified Commit c5afee40 authored by srihari-humbarwadi's avatar srihari-humbarwadi
Browse files

cast to original dtype after resizing

parent 177e2d71
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
official/vision/beta/projects/panoptic_maskrcnn/modeling/layers/fusion_layers.py
View file @ c5afee40
...@@ -142,6 +142,7 @@ class PanopticDeepLabFusion(tf.keras.layers.Layer): ...@@ -142,6 +142,7 @@ class PanopticDeepLabFusion(tf.keras.layers.Layer):
x = tf.image.resize( x = tf.image.resize(
x, size=[shape[1], shape[2]], x, size=[shape[1], shape[2]],
method=self._config_dict['interpolation']) method=self._config_dict['interpolation'])
x = tf.cast(x, dtype=feature.dtype)
x = tf.concat([x, feature], axis=self._channel_axis) x = tf.concat([x, feature], axis=self._channel_axis)
x = self._fusion_convs[i](x) x = self._fusion_convs[i](x)
......
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