aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/keras/applications/densenet_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/python/keras/applications/densenet_test.py')
-rw-r--r--tensorflow/python/keras/applications/densenet_test.py101
1 files changed, 0 insertions, 101 deletions
diff --git a/tensorflow/python/keras/applications/densenet_test.py b/tensorflow/python/keras/applications/densenet_test.py
deleted file mode 100644
index 8b6aa281ad..0000000000
--- a/tensorflow/python/keras/applications/densenet_test.py
+++ /dev/null
@@ -1,101 +0,0 @@
-# Copyright 2018 The TensorFlow Authors. All Rights Reserved.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-# http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-# ==============================================================================
-"""Tests for DenseNet application."""
-
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
-from tensorflow.python import keras
-from tensorflow.python.platform import test
-
-
-class DenseNet121Test(test.TestCase):
-
- def test_with_top(self):
- model = keras.applications.DenseNet121(weights=None)
- self.assertEqual(model.output_shape, (None, 1000))
-
- def test_no_top(self):
- model = keras.applications.DenseNet121(weights=None, include_top=False)
- self.assertEqual(model.output_shape, (None, None, None, 1024))
-
- def test_with_pooling(self):
- model = keras.applications.DenseNet121(weights=None,
- include_top=False,
- pooling='avg')
- self.assertEqual(model.output_shape, (None, 1024))
-
- def test_weight_loading(self):
- with self.assertRaises(ValueError):
- keras.applications.DenseNet121(weights='unknown',
- include_top=False)
- with self.assertRaises(ValueError):
- keras.applications.DenseNet121(weights='imagenet',
- classes=2000)
-
-
-class DenseNet169Test(test.TestCase):
-
- def test_with_top(self):
- model = keras.applications.DenseNet169(weights=None)
- self.assertEqual(model.output_shape, (None, 1000))
-
- def test_no_top(self):
- model = keras.applications.DenseNet169(weights=None, include_top=False)
- self.assertEqual(model.output_shape, (None, None, None, 1664))
-
- def test_with_pooling(self):
- model = keras.applications.DenseNet169(weights=None,
- include_top=False,
- pooling='max')
- self.assertEqual(model.output_shape, (None, 1664))
-
- def test_weight_loading(self):
- with self.assertRaises(ValueError):
- keras.applications.DenseNet169(weights='unknown',
- include_top=False)
- with self.assertRaises(ValueError):
- keras.applications.DenseNet169(weights='imagenet',
- classes=2000)
-
-
-class DenseNet201(test.TestCase):
-
- def test_with_top(self):
- model = keras.applications.DenseNet201(weights=None)
- self.assertEqual(model.output_shape, (None, 1000))
-
- def test_no_top(self):
- model = keras.applications.DenseNet201(weights=None, include_top=False)
- self.assertEqual(model.output_shape, (None, None, None, 1920))
-
- def test_with_pooling(self):
- model = keras.applications.DenseNet201(weights=None,
- include_top=False,
- pooling='avg')
- self.assertEqual(model.output_shape, (None, 1920))
-
- def test_weight_loading(self):
- with self.assertRaises(ValueError):
- keras.applications.DenseNet201(weights='unknown',
- include_top=False)
- with self.assertRaises(ValueError):
- keras.applications.DenseNet201(weights='imagenet',
- classes=2000)
-
-
-if __name__ == '__main__':
- test.main()