DenseNet

class DenseNet(num_init_features, growth_rate, block_config, bn_size=4, dropout=0, classes=1000, **kwargs)

Densenet-BC model from the “Densely Connected Convolutional Networks” 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 : list 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.