francisyizhang / densenet-keras

DenseNet Implementation in Keras with ImageNet Pretrained Models

DenseNet-Keras with ImageNet Pretrained Models

This is an Keras implementation of DenseNet with ImageNet pretrained weights. The weights are converted from Caffe Models. The implementation supports both Theano and TensorFlow backends.

To know more about how DenseNet works, please refer to the original paper

Densely Connected Convolutional Networks
Gao Huang, Zhuang Liu, Kilian Q. Weinberger, Laurens van der Maaten
arXiv:1608.06993

Pretrained DenseNet Models on ImageNet

The top-1/5 accuracy rates by using single center crop (crop size: 224x224, image size: 256xN)

Network	Top-1	Top-5	Theano	Tensorflow
DenseNet 121 (k=32)	74.91	92.19	model (32 MB)	model (32 MB)
DenseNet 169 (k=32)	76.09	93.14	model (56 MB)	model (56 MB)
DenseNet 161 (k=48)	77.64	93.79	model (112 MB)	model (112 MB)

Usage

First, download the above pretrained weights to the imagenet_models folder.

Run test_inference.py for an example of how to use the pretrained model to make inference.

python test_inference.py

Fine-tuning

Check this out to see example of fine-tuning DenseNet with your own dataset.

Requirements

• Keras 1.2.2 2.0.5
• Theano 0.8.2 or TensorFlow 0.12.0 1.2.1

Updates

• Keras 2.0.5 and TensorFlow 1.2.1 are supported