Commit eac1af65 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change

PiperOrigin-RevId: 469212225
parent 83911092
......@@ -670,7 +670,7 @@ class SegClassNet(tf.keras.layers.Layer):
self.min_level = min_level
self.max_level = max_level
self.fullres_output = fullres_output
self.fullres_conv_transpose = fullres_skip_connections
self.fullres_skip_connections = fullres_skip_connections
self.fnode = FNode(
0, # Always use the first level with highest resolution.
......@@ -726,7 +726,7 @@ class SegClassNet(tf.keras.layers.Layer):
if self.fullres_output:
for i in reversed(range(self.min_level)):
if self.config.fullres_skip_connections:
if self.fullres_skip_connections:
net = tf.keras.layers.Concatenate()([net, backbone_feats[i + 1]])
net = self.fullres_conv[str(i)](net)
net = self.fullres_conv_transpose[str(i)](net)
......
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