Skip to content

Instantly share code, notes, and snippets.

@backnotprop

backnotprop/convert.py

Created January 12, 2018 19:24
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 backnotprop/886580af9ebb7939c2457b0ed46dd60d to your computer and use it in GitHub Desktop.
Save backnotprop/886580af9ebb7939c2457b0ed46dd60d to your computer and use it in GitHub Desktop.
Download ZIP
Raw
convert.py
import tfcoreml as tf_converter
tf_converter.convert(tf_model_path = 'output_graph.pb',
mlmodel_path = 'model_name.mlmodel',
output_feature_names = ['add_37:0'],
## Note found this after running a conversion the first time
image_input_names = ['img_placeholder__0'])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment