NAME

            

              
              
AI::MXNet::Gluon::ModelZoo::Vision::DenseNet - Densenet-BC model from the "Densely Connected Convolutional Networks"
            
          

DESCRIPTION

            

              
              
Densenet-BC model from the "Densely Connected Convolutional Networks" <https://arxiv.org/pdf/1608.06993.pdf> paper.
Parameters
----------
num_init_features : Int
    Number of filters to learn in the first convolution layer.
growth_rate : Int
    Number of filters to add each layer (`k` in the paper).
block_config : array ref of Int
    List of integers for numbers of layers in each pooling block.
bn_size : Int, default 4
    Multiplicative factor for number of bottle neck layers.
    (i.e. bn_size * k features in the bottleneck layer)
dropout : float, default 0
    Rate of dropout after each dense layer.
classes : int, default 1000
    Number of classification classes.
            
          

get_densenet

            

              
              
Densenet-BC model from the
"Densely Connected Convolutional Networks" <https://arxiv.org/pdf/1608.06993.pdf> paper.
Parameters
----------
$num_layers : Int
    Number of layers for the variant of densenet. Options are 121, 161, 169, 201.
:$pretrained : Bool, default 0
    Whether to load the pretrained weights for model.
:$ctx : AI::MXNet::Context, default CPU
    The context in which to load the pretrained weights.
:$root : Str, default '~/.mxnet/models'
    Location for keeping the model parameters.
            
          

densenet121

            

              
              
Densenet-BC 121-layer model from the
"Densely Connected Convolutional Networks" <https://arxiv.org/pdf/1608.06993.pdf> paper.
Parameters
----------
:$pretrained : Bool, default 0
    Whether to load the pretrained weights for model.
:$ctx : AI::MXNet::Context, default CPU
    The context in which to load the pretrained weights.
:$root : Str, default '~/.mxnet/models'
    Location for keeping the model parameters.
            
          

densenet161

            

              
              
Densenet-BC 161-layer model from the
"Densely Connected Convolutional Networks" <https://arxiv.org/pdf/1608.06993.pdf> paper.
Parameters
----------
:$pretrained : Bool, default 0
    Whether to load the pretrained weights for model.
:$ctx : AI::MXNet::Context, default CPU
    The context in which to load the pretrained weights.
:$root : Str, default '~/.mxnet/models'
    Location for keeping the model parameters.
            
          

densenet169

            

              
              
Densenet-BC 169-layer model from the
"Densely Connected Convolutional Networks" <https://arxiv.org/pdf/1608.06993.pdf> paper.
Parameters
----------
:$pretrained : Bool, default 0
    Whether to load the pretrained weights for model.
:$ctx : AI::MXNet::Context, default CPU
    The context in which to load the pretrained weights.
:$root : Str, default '~/.mxnet/models'
    Location for keeping the model parameters.
            
          

densenet201

            

              
              
Densenet-BC 201-layer model from the
"Densely Connected Convolutional Networks" <https://arxiv.org/pdf/1608.06993.pdf> paper.
Parameters
----------
:$pretrained : Bool, default 0
    Whether to load the pretrained weights for model.
:$ctx : AI::MXNet::Context, default CPU
    The context in which to load the pretrained weights.
:$root : Str, default '~/.mxnet/models'
    Location for keeping the model parameters.