Add squeezenet option with weights wrapped in this package

datarobot / pic2vec

Lightweight Image Featurization Made Easy

Other

31 stars 14 forks source link

Add squeezenet option with weights wrapped in this package #4

Closed zachmayer closed 7 years ago

zachmayer commented 7 years ago

Adding squeezenet weights to the package gives us a fallback for cases where it's not possible to download the weights for the other networks.

https://github.com/rcmalli/keras-squeezenet

zachmayer commented 7 years ago

Different squeezenet: https://github.com/wohlert/keras-squeezenet https://github.com/wohlert/keras-squeezenet/blob/master/squeezenet_weights.h5

joristaglio commented 7 years ago

Done! Added rcmalli's version and set SqueezeNet as the default network, loaded from weights. It obtains 94% accuracy on the Cats vs. Dogs dataset after just slapping a linear classifier on top.