Introduction

Tutorials

Developer guide

API Reference

private densenetfunction

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 block
  • growth_rates: the growth rates of output feature maps within each dense_block (a vector of vectors)
  • reduction: the factor by which the number of feature maps is scaled across each transition
  • nclasses: 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 transitions
  • growth_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 transition
  • nclasses: the number of output classes