tf.keras.applications.DenseNet121( include_top=True, weights='imagenet', input_tensor=None, input_shape=None, pooling=None, classes=1000 )
Instantiates the DenseNet architecture.
Optionally loads weights pre-trained on ImageNet. Note that when using TensorFlow, for best performance you should set
image_data_format='channels_last' in your Keras config at ~/.keras/keras.json.
The model and the weights are compatible with TensorFlow, Theano, and CNTK. The data format convention used by the model is the one specified in your Keras config file.
blocks: numbers of building blocks for the four dense layers.
include_top: whether to include the fully-connected layer at the top of the network.
weights: one of
None(random initialization), 'imagenet' (pre-training on ImageNet), or the path to the weights file to be loaded.
input_tensor: optional Keras tensor (i.e. output of
layers.Input()) to use as image input for the model.
input_shape: optional shape tuple, only to be specified if
include_topis False (otherwise the input shape has to be
(224, 224, 3)(with
channels_lastdata format) or
(3, 224, 224)(with
channels_firstdata format). It should have exactly 3 inputs channels.
pooling: optional pooling mode for feature extraction when
Nonemeans that the output of the model will be the 4D tensor output of the last convolutional layer. -
avgmeans that global average pooling will be applied to the output of the last convolutional layer, and thus the output of the model will be a 2D tensor. -
maxmeans that global max pooling will be applied.
classes: optional number of classes to classify images into, only to be specified if
include_topis True, and if no
weightsargument is specified.
A Keras model instance.
ValueError: in case of invalid argument for
weights, or invalid input shape.
© 2018 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.