Add resnet50 test.
@@ -4,6 +4,26 @@ from keras import applications
from keras import backend as K
+@keras_test
+def test_resnet50():
+ model = applications.ResNet50(weights=None)
+ assert model.output_shape == (None, 1000)
+
+
+@keras_test
+def test_resnet50_notop():
+ model = applications.ResNet50(weights=None, include_top=False)
+ assert model.output_shape == (None, None, None, 2048)
+
+
+@keras_test
+def test_resnet50_pooling():
+ model = applications.ResNet50(weights=None,
+ include_top=False,
+ pooling='avg')
+ assert model.output_shape == (None, 2048)
+
+
@keras_test
def test_vgg16():
model = applications.VGG16(weights=None)
