如何将resnet50转换为coreml模型?

时间:2019-06-03 10:19:47

标签: tensorflow coreml eager-execution

如何将以eager_execution模式编写的resnet50网络转换为coreml?

这是tf resnet50的实现: https://github.com/tensorflow/tensorflow/blob/r1.13/tensorflow/contrib/eager/python/examples/resnet50/resnet50.py

1 个答案:

答案 0 :(得分:0)

在这一点上,coremltools和tfcoreml不处理紧急模式。您需要先将TF模型转换为“冻结的” .pb图形文件,然后才能使用tfcoreml将其转换为Core ML。

如果无法将模型转换为冻结的图形文件,则不能使用自动转换工具...但是,即使在那种情况下,仍然可以使用NeuralNetworkBuilder(coremltools的一部分)创建Core ML模型。 ,但您必须手动完成所有这些操作。