Skip to content

A keras implementation of DenseNet

Notifications You must be signed in to change notification settings

parthghughri/densenet

 
 

Repository files navigation

densenet

This project includes:

  • densenet.py, a keras implementation of DenseNet.

  • cifar10-test.py, code to train the model on Cifar 10, also to test the trained model.

  • vis_densenet.py, a demo to visualize the layers of the model, using keras-vis.

  • testnet.py, a vgg-like model, a convnet model with bottle-neck layers and densenet-style skip connections.

References

  1. Gao Huang, Zhuang Liu and K. Weinberger. Densely Connected Convolutional Networks. Arxiv. Github.

  2. DenseNet Other implementations, especially, copied Christopher Masch's code and fixed some issues for aligning to the original DenseNet paper.

About

A keras implementation of DenseNet

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%