Chainer squeezenet model
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
README.md
model.py

README.md

SqueezeNet

An implementation of SqueezeNet in chainer.

Below are some benchmarks on a ImageNet-like dataset (1 million 255x255 images with 128 batch size)

Model Size (mb) Parameters (million) Accuracy
ZFNet 117 16.42 0.5835
SqueezeNet 4.6 1.288 0.5207

Details

  • ZFNet instead of AlexNet (I already had a trained ZFNet model).
  • This is the basic squeezenet without deep compression. I had to add a dense layer to the end of SqueezeNet to get correct shape for my labels. I also had to add batch normalisation to the fire modules so it would fit.
  • The paper says images are 224x224 but the code and parameters suggests they use 227x227. I also added some padding, relu and initialisation that was used in the squeezenet code but not mentioned in the paper.
  • Accuracy of squeezenet is lower in this test but there is a 25x size reduction over ZFNet (SqueezeNet size is the same as in the paper) and a 12x reduction in params.