Skip to content

转换为 onnx 模型该如何写 outputs 和 inputs #105

@lusi1990

Description

@lusi1990

你好, 我参考文章 验证码识别新革命:源码+通用识别模型 已经成功训练出了模型, 想使用 tf2onnx 转换为 onnx
命令如下

python -m tf2onnx.convert --input out/graph/are-CNN5-NoRecurrent-H64-CrossEntropy-C1_9033.pb  --inputs input:0 --outputs output:0  --output model.onnx

然后报错提示 output is not in graph

查找资料后应该是需要 配置 --inputs --outputs , 请问该如何配置该参数?

我也查看了 tf_onnx_util.py 文件, 仍然不知道如何调用?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions