为onnx模型开发阅读器

时间:2018-01-27 08:49:21

标签: julia protocol-buffers models

我正在尝试为朱莉娅的onnx模型做一个读者。通常,onnx模型由model.pb文件组成,其中包含有关模型的元数据。但是,我无法弄清楚如何使用协议缓冲区或Julia实现协议缓冲区(ProtoBuf.jl)从此model.pb文件生成数据结构,以便我以后可以为此生成Julia代码模型。

有人能告诉我如何让读者将onnx模型加载到Julia中?

1 个答案:

答案 0 :(得分:0)

万一有人需要,我找到了答案:使用ProtoBuf.jl将ONNX模型读入ModelProto结构非常简单:

using ProtoBuf f = readproto(open("model.onnx"), Proto.ModelProto())

f是对应的ONNX图。