Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
You can not select more than 25 topics
Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tammy Yang
616cb8db09
|
6 years ago | |
|---|---|---|
| LICENSE.md | 8 years ago | |
| README.md | 6 years ago | |
| model.png | 8 years ago | |
| model.py | 6 years ago | |
| squeezenet_demo.py | 7 years ago | |
README.md
SqueezeNet Keras Implementation
This is the Keras implementation of SqueezeNet using functional API (arXiv 1602.07360).
SqueezeNet is a small model of AlexNet-level accuracy with 50x fewer parameters and <0.5MB model size.
The original model was implemented in caffe.
Reference
Differences:
- Switch from Graph model to Keras 1.0 functional API
- Fix the bug of pooling layer
- Many thanks to StefOe, the source can now support Keras 2.0 API.
Result
This repository contains only the Keras implementation of the model, for other parameters used, please see the demo script, squeezenet_demo.py in the simdat package.
The training process uses a total of 2,600 images with 1,300 images per class (so, total two classes only).
There are a total 130 images used for validation. After 20 epochs, the model achieves the following:
loss: 0.6563 - acc: 0.7065 - val_loss: 0.6247 - val_acc: 0.8750
Model Visualization
