Skip to content

DenseNet Implementation in Keras with ImageNet Pretrained Models

License

Notifications You must be signed in to change notification settings

flyyufelix/DenseNet-Keras

Repository files navigation

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

About

DenseNet Implementation in Keras with ImageNet Pretrained Models

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages