TensorFlow 训练出来的模型在 iOS12 上怎么运行?

目前的流程如下:

1、ckpt freeze 固化模型生成 pb

TensorFlow 训练出来的模型在 iOS12 上怎么运行?_第1张图片

 

2  pb 通过 tf-coreml 转成 mlmodel

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'],
                     image_input_names = 'input:0',
                     class_labels = 'imagenet_slim_labels.txt',
                     red_bias = -1,
                     green_bias = -1,
                     blue_bias = -1,
                     image_scale = 2.0/255.0
                     )

3  用 coreml 修改了下生成 mlmodel 的输出类型

 

 

TensorFlow 训练出来的模型在 iOS12 上怎么运行?

 

 

 

 

 

 

 

 

你可能感兴趣的:(Deep,Learning)