search

datarobot / pic2vec

Lightweight Image Featurization Made Easy
Other
31 stars 14 forks source link

Add support for all pretrained keras networks #5

Closed zachmayer closed 7 years ago

zachmayer commented 7 years ago

Since we're downloading weights anyways, no reason not to allow the user to choose any of them. Keep the default as one of the inception networks that works on both theanon and tensorflow.

zachmayer commented 7 years ago

Might also make sense to use squeezenet as the default: https://github.com/datarobot/imagefeaturizer/issues/4

Then there's now downloading required at all for a quick start.

joristaglio commented 7 years ago

Done! Added squeezenet with weights packaged into the network, and set it as the default. All models can be called in the ImageFeaturizer() initialization with the model='___' option.