DenseNet implementation using Tensorflow 2

Instalation

$ pip install densenet

Usage

from densenet import densenet_model
model = densenet_model(classes=n_clases)

you can disable the se layers by setting the argument with_se_layers to false.

Changing nb_layers lenght will change the number of dense layers.

References

Inspired by flyyufelix keras implementation (https://github.com/flyyufelix/DenseNet-Keras).

For more information about densenet please refer to the original paper (https://arxiv.org/abs/1608.06993).