Skip to content

GitLab

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

Internal change 

PiperOrigin-RevId: 449359795
parent cdf03125
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
official/vision/modeling/heads/segmentation_heads_test.py
View file @ c0b12110
...@@ -63,7 +63,7 @@ class SegmentationHeadTest(parameterized.TestCase, tf.test.TestCase): ...@@ -63,7 +63,7 @@ class SegmentationHeadTest(parameterized.TestCase, tf.test.TestCase):
logits = head((backbone_features, decoder_features)) logits = head((backbone_features, decoder_features))
if level in decoder_features: if str(level) in decoder_features:
self.assertAllEqual(logits.numpy().shape, [ self.assertAllEqual(logits.numpy().shape, [
2, decoder_features[str(level)].shape[1], 2, decoder_features[str(level)].shape[1],
decoder_features[str(level)].shape[2], 10 decoder_features[str(level)].shape[2], 10
...@@ -94,6 +94,7 @@ class MaskScoringHeadTest(parameterized.TestCase, tf.test.TestCase): ...@@ -94,6 +94,7 @@ class MaskScoringHeadTest(parameterized.TestCase, tf.test.TestCase):
num_convs=num_convs, num_convs=num_convs,
num_filters=num_filters, num_filters=num_filters,
fc_dims=128, fc_dims=128,
num_fcs=num_fcs,
fc_input_size=fc_input_size) fc_input_size=fc_input_size)
scores = head(features) scores = head(features)
......
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