Skip to content

GitLab

Commit 13c8470b authored by Abdullah Rashwan's avatar Abdullah Rashwan Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 356906020
parent bc324fda
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
official/vision/beta/modeling/heads/segmentation_heads.py
View file @ 13c8470b
...@@ -182,6 +182,7 @@ class SegmentationHead(tf.keras.layers.Layer): ...@@ -182,6 +182,7 @@ class SegmentationHead(tf.keras.layers.Layer):
x = tf.image.resize( x = tf.image.resize(
x, tf.shape(y)[1:3], method=tf.image.ResizeMethod.BILINEAR) x, tf.shape(y)[1:3], method=tf.image.ResizeMethod.BILINEAR)
x = tf.cast(x, dtype=y.dtype)
x = tf.concat([x, y], axis=self._bn_axis) x = tf.concat([x, y], axis=self._bn_axis)
elif self._config_dict['feature_fusion'] == 'pyramid_fusion': elif self._config_dict['feature_fusion'] == 'pyramid_fusion':
x = nn_layers.pyramid_feature_fusion(decoder_output, x = nn_layers.pyramid_feature_fusion(decoder_output,
......
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