0

我有一个 onnx 图形/模型,其中有很大的常量,因此加载和解析它需要很多时间。我可以从图中“剥离”数据,以便在没有数据的情况下检查图节点吗?

4

1 回答 1

0

Initializer 是 GraphProto 中的字段之一。您应该能够使用简单的 python 脚本清除初始化字段。我没有测试以下代码,但应该是这样的:

import onnx

def clear_initializer(model_path):
    model = onnx.load_model(model_path)
    model.graph.ClearField('initializer')
    onnx.save_model(model)

参考: https : //developers.google.com/protocol-buffers/docs/reference/python/google.protobuf.message.Message-class https://github.com/onnx/onnx/blob/2e7099ee7c37b196c197c9a084a97698a41da232/onnx/init .py

于 2018-12-03T21:10:09.460 回答