Home | History | Annotate | Download | only in applications
      1 # Copyright 2018 The TensorFlow Authors. All Rights Reserved.
      2 #
      3 # Licensed under the Apache License, Version 2.0 (the "License");
      4 # you may not use this file except in compliance with the License.
      5 # You may obtain a copy of the License at
      6 #
      7 #     http://www.apache.org/licenses/LICENSE-2.0
      8 #
      9 # Unless required by applicable law or agreed to in writing, software
     10 # distributed under the License is distributed on an "AS IS" BASIS,
     11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     12 # See the License for the specific language governing permissions and
     13 # limitations under the License.
     14 # ==============================================================================
     15 # pylint: disable=invalid-name
     16 """DenseNet models for Keras.
     17 """
     18 from __future__ import absolute_import
     19 from __future__ import division
     20 from __future__ import print_function
     21 
     22 from keras_applications import densenet
     23 
     24 from tensorflow.python.keras.applications import keras_modules_injection
     25 from tensorflow.python.util.tf_export import keras_export
     26 
     27 
     28 @keras_export('keras.applications.densenet.DenseNet121',
     29               'keras.applications.DenseNet121')
     30 @keras_modules_injection
     31 def DenseNet121(*args, **kwargs):
     32   return densenet.DenseNet121(*args, **kwargs)
     33 
     34 
     35 @keras_export('keras.applications.densenet.DenseNet169',
     36               'keras.applications.DenseNet169')
     37 @keras_modules_injection
     38 def DenseNet169(*args, **kwargs):
     39   return densenet.DenseNet169(*args, **kwargs)
     40 
     41 
     42 @keras_export('keras.applications.densenet.DenseNet201',
     43               'keras.applications.DenseNet201')
     44 @keras_modules_injection
     45 def DenseNet201(*args, **kwargs):
     46   return densenet.DenseNet201(*args, **kwargs)
     47 
     48 
     49 @keras_export('keras.applications.densenet.decode_predictions')
     50 @keras_modules_injection
     51 def decode_predictions(*args, **kwargs):
     52   return densenet.decode_predictions(*args, **kwargs)
     53 
     54 
     55 @keras_export('keras.applications.densenet.preprocess_input')
     56 @keras_modules_injection
     57 def preprocess_input(*args, **kwargs):
     58   return densenet.preprocess_input(*args, **kwargs)
     59