Skip to content

GitLab

Commit db87dd5b authored by Francois Chollet's avatar Francois Chollet Committed by TF Object Detection Team
Browse files

Enable input spec checking for Functional models. 

PiperOrigin-RevId: 324625967
parent d5776810
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
research/object_detection/models/keras_models/resnet_v1_tf2_test.py
View file @ db87dd5b
...@@ -202,9 +202,9 @@ class ResnetShapeTest(test_case.TestCase, parameterized.TestCase): ...@@ -202,9 +202,9 @@ class ResnetShapeTest(test_case.TestCase, parameterized.TestCase):
}) })
def test_output_shapes(self, resnet_type, output_layer_names): def test_output_shapes(self, resnet_type, output_layer_names):
if resnet_type == 'resnet_v1_34': if resnet_type == 'resnet_v1_34':
model = resnet_v1.resnet_v1_34(weights=None) model = resnet_v1.resnet_v1_34(input_shape=(64, 64, 3), weights=None)
else: else:
model = resnet_v1.resnet_v1_18(weights=None) model = resnet_v1.resnet_v1_18(input_shape=(64, 64, 3), weights=None)
outputs = [ outputs = [
model.get_layer(output_layer_name).output model.get_layer(output_layer_name).output
for output_layer_name in output_layer_names for output_layer_name in output_layer_names
......
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