public DenseNet — struct
DenseNet(nblocks::NTuple{N, <:Integer};
growth_rate = 32, reduction = 0.5, nclasses = 1000)
Create a DenseNet model
(reference).
See also densenet.
Arguments
nblocks: number of dense blocks between transitionsgrowth_rate: the output feature map growth rate of dense blocks (i.e.kin the paper)reduction: the factor by which the number of feature maps is scaled across each transitionnclasses: the number of output classes
DenseNet(config::Integer = 121; pretrain = false, nclasses = 1000)
DenseNet(transition_config::NTuple{N,Integer})
Create a DenseNet model with specified configuration. Currently supported values are (121, 161, 169, 201)
(reference).
Set pretrain = true to load the model with pre-trained weights for ImageNet.
Warning
DenseNet does not currently support pretrained weights.
See also Metalhead.densenet.