Skip to content

Instantly share code, notes, and snippets.

@caisq

caisq/keras_mobilenet_to_tfjs.py

Created October 27, 2018 17:00
Show Gist options
  • Save caisq/c1e9470794ab15c5c840e7f082cd8ccb to your computer and use it in GitHub Desktop.
Save caisq/c1e9470794ab15c5c840e7f082cd8ccb to your computer and use it in GitHub Desktop.
Download ZIP
Convert Keras MobileNet Application intothe TensorFlow.js Format
Raw
keras_mobilenet_to_tfjs.py
import keras
import tensorflowjs as tfjs
model = keras.applications.mobilenet.MobileNet(alpha=0.25)
tfjs.converters.save_keras_model(model, '/tmp/mobilnet_0.25')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment