https://github.com/liuzhuang13/DenseNetCaffe

Caffe code for Densely Connected Convolutional Networks (DenseNets)
https://github.com/liuzhuang13/DenseNetCaffe

Last synced: 24 days ago
JSON representation

Caffe code for Densely Connected Convolutional Networks (DenseNets)

• Host: GitHub
• URL: https://github.com/liuzhuang13/DenseNetCaffe
• Owner: liuzhuang13
• Created: 2016-09-05T03:58:02.000Z (over 8 years ago)
• Default Branch: master
• Last Pushed: 2017-08-11T21:47:41.000Z (over 7 years ago)
• Last Synced: 2024-11-11T03:05:47.416Z (about 1 month ago)
• Language: Python
• Size: 10.7 KB
• Stars: 269
• Watchers: 21
• Forks: 208
• Open Issues: 9
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

• awesome-very-deep-learning - Caffe Implementation

README

        
# Densely Connected Convolutional Network (DenseNet)

This repository contains the caffe version code for the paper [Densely Connected Convolutional Networks](http://arxiv.org/abs/1608.06993). 

For a brief introduction of DenseNet, see our original [Torch implementation](https://github.com/liuzhuang13/DenseNet).

## ImageNet Pretrained Models

See https://github.com/shicai/DenseNet-Caffe for caffe prototxt and pre-trained models.

See https://github.com/liuzhuang13/DenseNet for Torch pre-trained models.

See http://pytorch.org/docs/torchvision/models.html?highlight=densenet for directly using the pretrained models in PyTorch.

## Note

0. The models in this repo are for CIFAR datasets only (input 32x32). If you feed images with larger resolution (e.g., ImageNet images), you need to use a different downsampling strategy to keep the memory usage reasonable. See our paper or Torch code for details on ImageNet models.

0. The code in this repo doesn't support BC-structres. However, it should be easy to modify.

0. This code is not the code we use to obtain the results in the original paper, the details (such as input preprocessing, data augmentation, training epochs) may be different. To reproduce the results reported in our paper, see our original [Torch implementation](https://github.com/liuzhuang13/DenseNet).

## Results

The default setting (L=40, k=12, dropout=0.2) in the code yields a 7.09% error rate on CIFAR10 dataset (without any data augmentation).

## Usage 

0. Get the CIFAR data prepared following the [Caffe's official CIFAR tutorial](http://caffe.berkeleyvision.org/gathered/examples/cifar10.html).

1. make\_densenet.py contains the code to generate the network and solver prototxt file. First change the data path in function make\_net() and preprocessing mean file in function densenet() to your own path of corresponding data file.

2. By default make\_densenet.py generates a DenseNet with Depth L=40, Growth rate k=12 and Dropout=0.2. To experiment with different settings, change the code accordingly (see the comments in the code). Example prototxt files are already included. Use ```python densenet_make.py``` to generate new prototxt files.

3. Change the caffe path in train.sh. Then use ```sh train.sh``` to train a DenseNet.

## Contact

liuzhuangthu at gmail.com  

gh349 at cornell.edu   

Any discussions, suggestions and questions are welcome!