Skip to content

Instantly share code, notes, and snippets.

@francoismarceau29

francoismarceau29/convert-keras-to-core-ml.py

Created July 25, 2017 23:51
Show Gist options
  • Save francoismarceau29/6cc651827a753bd6587411628b344a32 to your computer and use it in GitHub Desktop.
Save francoismarceau29/6cc651827a753bd6587411628b344a32 to your computer and use it in GitHub Desktop.
Download ZIP
Convert Convolutional Neural Network Keras model to CoreML model (.mlmodel)
Raw
convert-keras-to-core-ml.py
import coremltools
coreml_model = coremltools.converters.keras.convert('model.h5', input_names='data', image_input_names='data', is_bgr=True, output_names='species')
coreml_model.save('model.mlmodel')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment