6.3. TopsIDEAS onnx opset

描述

转化onnx模型的opset。

命令行

使用方法

usage: topsideas onnx opset [-h] --input_onnx INPUT_ONNX [--output_onnx OUTPUT_ONNX] [--opset OPSET] [--keep_init_inputs]

参数

表 6.3.1 topsideas onnx opset 参数列表

short	long	default	help
-h	--help		show this help message and exit
	--input_onnx	None	Provide the original onnx file.
	--output_onnx	None	Export the modified onnx file.
	--opset	13	Set opset version.
	--keep_init_inputs		If 'keep_init_inputs' is True, it is enabled to keep init inputs when ir_version is greater than 4.

示例

topsideas onnx opset --input_onnx=demo.onnx --opset=13

图 6.3.1 onnx opset调整示例

API

使用方法

from topsideas.onnx import Opset
result = Opset.run(input_mp, opset)

参数

表 6.3.2 topsideas.onnx.Opset 参数列表

argument	type	default
input_mp	onnx.onnx_ml_pb2.ModelProto
opset	int
keep_init_inputs		False
RETURN	onnx.onnx_ml_pb2.ModelProto