【发布时间】:2020-10-30 18:38:17
【问题描述】:
tensorflow 模型中的输出层名称已自定义,但当我尝试使用 CoreML 对其进行转换时,它无法识别该层,除非我在尝试分配 output_description 时使用“Identity”。
【问题讨论】:
标签: tensorflow keras coreml
tensorflow 模型中的输出层名称已自定义,但当我尝试使用 CoreML 对其进行转换时,它无法识别该层,除非我在尝试分配 output_description 时使用“Identity”。
【问题讨论】:
标签: tensorflow keras coreml
您可以在 Python 脚本中使用 coremltools 加载 mlmodel 文件,然后使用 rename_feature 函数重命名输出。
【讨论】: