bitorch.models.densenet

Functions

basedensenet_constructor

Creates a densenet of the given model type with given layer numbers.

Classes

BaseNetDense(num_init_features, growth_rate, ...)

Densenet-BC model from the "Densely Connected Convolutional Networks" paper.

DenseLayer(num_features, growth_rate, ...)

DenseNet(num_layers, input_shape[, ...])

DenseNet28(*args, **kwargs)

BinaryDenseNet-28 model from "BinaryDenseNet: Developing an Architecture for Binary Neural Networks" paper.

DenseNet37(*args, **kwargs)

BinaryDenseNet-37 model from "BinaryDenseNet: Developing an Architecture for Binary Neural Networks" paper.

DenseNet45(*args, **kwargs)

BinaryDenseNet-45 model from "BinaryDenseNet: Developing an Architecture for Binary Neural Networks" paper.

DenseNetFlex(*args, **kwargs)

Flexible BinaryDenseNet model from "BinaryDenseNet: Developing an Architecture for Binary Neural Networks" <https://openaccess.thecvf.com/content_ICCVW_2019/html/NeurArch/Bethge_BinaryDenseNet_Developing_an_Architecture_for_Binary_Neural_Networks_ICCVW_2019_paper.html> paper.