Introduction

Tutorials

Developer guide

API Reference

public DenseNetstruct

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 transitions
  • growth_rate: the output feature map growth rate of dense blocks (i.e. k in the paper)
  • reduction: the factor by which the number of feature maps is scaled across each transition
  • nclasses: 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.