我能否将Google的Tensorflow上创建的".py“或".ipynb”模型/文件转换为coreML模型/文件？
提问于 2019-09-24 00:56:29
我在Google的Tensorflow上创建了一个CNN模型，并以".py“文件的形式下载了该文件。如何将此文件转换为".coreml“文件？
任何建议都非常感谢！
谢谢!
回答 3
Stack Overflow用户
回答已采纳
发布于 2019-09-24 08:14:14
关于如何转换为核心ML的.mlmodel：https://developer.apple.com/documentation/coreml/converting_trained_models_to_core_ml的文档
对于TensorFlow模型，需要将其转换为frozen.pb模型作为转换函数的输入。下面是一些示例：https://github.com/tf-coreml/tf-coreml/tree/master/examples
Stack Overflow用户
发布于 2019-09-25 03:30:03
如果您已经下载了一个.py文件，那么您没有下载模型，而是下载了整个Python。
在编译模型时，可以使用python包coremltools将TensorFlow/Keras模型转换为.mlmodel文件。
import coremltools
core_mlmodel = coremltools.converters.keras.convert(your_keras_model, respect_trainable=True)
core_mlmodel.save("YourModel.mlmodel")然后，您将在Files中左侧找到您的模型。
Stack Overflow用户
发布于 2019-10-03 19:52:57
@ Sebastian J. Vogt.这是我的转换代码。这是正确的吗？谢谢!
import coremltools
category_output_labels = ['dress', 'shirt', 'trousers']
color_output_labels = ['black', 'blue', 'green', 'pink', 'red',
'white']
coreml_model = coremltools.converters.keras.convert('./fashion_multi_output_v02.h5',输入_名称=‘图像’，图像_输入_名称=‘图像’，类_标签=‘类别_输出_标签’，'color_output_labels')
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/58076574
复制相关文章
点击加载更多
腾讯云开发者
