如何正确使用tf-coreml？

Question

我想将张量流模型（https://github.com/vanhuyz/CycleGAN-TensorFlow）转换为coreml模型，以便能够在iOS上导入它。

我知道执行转换只需使用以下代码即可：

import tfcoreml as tf_converter

tf_converter.convert(tf_model_path = 'my_model.pb', 
                     mlmodel_path = 'my_model.mlmodel', 
                     output_feature_names = ['softmax:0'])

但是我不知道在第三个参数“ output_feature_names”中应输入什么。有想法吗？

Answer 1

如果使用Netron打开.pb文件并一直滚动到底部，您会看到最后一层是EncodeJpeg，因此输出要素名称将是EncodeJpeg:0。

但是，Core ML没有“ encode jpeg”操作，因此您不能转换整个图形。您需要找到Core ML仍支持的最后一个张量。

我会尝试output_feature_names=["G_7/output/Tanh:0"]，但是图形中还有很多其他东西无法转换。