前言

對(duì)已有的onnx結(jié)構(gòu)，進(jìn)行簡(jiǎn)化操作，onnx提供兩種常規(guī)操作

方式一

假設(shè)為 model.onnx, 比較簡(jiǎn)單粗暴

# 簡(jiǎn)化
onnxsim model.onnx model_sim.onnx

方式二

稍微復(fù)雜點(diǎn)，代碼有點(diǎn)多

import onnx
import argparse
from onnxsim import simplify# Simplify
def simplify_model(args):onnx_model = onnx.load(args.origin_model)model_simp, check = simplify(onnx_model)model_simp = onnx.shape_inference.infer_shapes(model_simp)onnx.save(model_simp, args.output_model)print("  Simplify onnx Done.")# 檢查onnx計(jì)算圖
def checknet(model_path):model = onnx.load(model_path)onnx.checker.check_model(model)  # Print a human readable representation of the graph# print(onnx.helper.printable_graph(model.graph))def parse_args():parser = argparse.ArgumentParser()parser.add_argument("--origin_model", type=str)parser.add_argument("--output_model", type=str)args = parser.parse_args()return argsif __name__ == "__main__":args = parse_args()simplify_model(args

總結(jié)

• 兩種本質(zhì)上沒(méi)用改變，都是通過(guò)調(diào)用onnxsim進(jìn)行操作
• 看個(gè)人使用時(shí)機(jī)