如何将 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
函数重命名输出。
中的文档
tensorflow 模型中的输出层名称已自定义,但当我尝试使用 CoreML 对其进行转换时,它将无法识别该层,除非我在尝试分配 output_description
.
您可以在 Python 脚本中使用 coremltools 加载 mlmodel 文件,然后使用 rename_feature
函数重命名输出。