tensorflow - 如何将 resnet50 转换为 coreml 模型？

Question

如何将eager_execution模式编写的resnet50网络转成coreml？

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

Answer 1

此时，coremltools 和 tfcoreml 不处理 Eager 模式。您需要先将 TF 模型转换为“冻结”的 .pb 图形文件，然后您可以使用 tfcoreml 将其转换为 Core ML。

如果你不能将你的模型变成一个冻结的图形文件，你就不能使用自动转换工具......但即使在那种情况下，你仍然可以使用 NeuralNetworkBuilder（coremltools 的一部分）创建一个 Core ML 模型，但是你'将不得不手工完成所有这些。