/ 
AI-MXNet-Gluon-ModelZoo-1.33
River stage zero No dependents
/ AI::MXNet::Gluon::ModelZoo::Vision::DenseNet

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.