Skip to content

Instantly share code, notes, and snippets.

@jasonsalas

jasonsalas/get_model.py

Created October 25, 2019 23:10
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save jasonsalas/f18bf4924dab249215ad8eb394646d09 to your computer and use it in GitHub Desktop.
Save jasonsalas/f18bf4924dab249215ad8eb394646d09 to your computer and use it in GitHub Desktop.
Download ZIP
Download pre-trained VGG16 model on ImageNet weights
Raw
get_model.py
from keras.applications.vgg16 import VGG16
model = VGG16(weights='imagenet')
model.save('model_vgg16_imagenet.h5')
print('Pre-trained VGG16 model with ImageNet weights saved!')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment