如何将 CoreML 模型中的输出名称从 Identity 更改为其他名称?

How do I change the ouput's name in a CoreML model from Identity to something else?

tensorflow 模型中的输出层名称已自定义,但当我尝试使用 CoreML 对其进行转换时,它将无法识别该层,除非我在尝试分配 output_description.

您可以在 Python 脚本中使用 coremltools 加载 mlmodel 文件,然后使用 rename_feature 函数重命名输出。

请参阅 https://coremltools.readme.io/docs/mlmodel-utilities

中的文档