Skip to content

Instantly share code, notes, and snippets.

@azamsharp
Created December 18, 2017 20:17
Show Gist options
  • Save azamsharp/4a8e4e0f2c277a34c7782dfe793b4882 to your computer and use it in GitHub Desktop.
Save azamsharp/4a8e4e0f2c277a34c7782dfe793b4882 to your computer and use it in GitHub Desktop.
convertor output feature names
import tfcoreml as tf_converter
tf_converter.convert(tf_model_path = 'inception_v1_2016_08_28_frozen.pb',
mlmodel_path = 'InceptionV1.mlmodel',
output_feature_names = ['InceptionV1/Logits/Predictions/Softmax:0'])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment