private
densenet
— function
densenet(inplanes, growth_rates; reduction = 0.5, nclasses = 1000)
Create a DenseNet model (reference).
Arguments
inplanes
: the number of input feature maps to the first dense blockgrowth_rates
: the growth rates of output feature maps within eachdense_block
(a vector of vectors)reduction
: the factor by which the number of feature maps is scaled across each transitionnclasses
: the number of output classes
densenet(nblocks; growth_rate = 32, reduction = 0.5, nclasses = 1000)
Create a DenseNet model (reference).
Arguments
nblocks
: number of dense blocks between transitionsgrowth_rate
: the output feature map growth rate of dense blocks (i.e.k
in the ref)reduction
: the factor by which the number of feature maps is scaled across each transitionnclasses
: the number of output classes