Skip to content

GitLab

"...ops/git@developer.sourcefind.cn:OpenDAS/mmdetection3d.git" did not exist on "16c3f6e18b695cac403afd1345a30c2bc3ae88b6"
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