4.3 cut onnx结构

最新推荐文章于 2024-07-18 23:16:17 发布

nsq_ai

最新推荐文章于 2024-07-18 23:16:17 发布

阅读量148

点赞数

分类专栏：模型推理文章标签：性能优化 onnx

本文链接：https://blog.csdn.net/weixin_45063703/article/details/134372682

版权

模型推理专栏收录该内容

54 篇文章 3 订阅

订阅专栏

文章介绍了如何使用Python的onnx库对ONNX模型进行部分结构提取，提到了onnx.utils.extract_model函数的应用，同时指出某些情况下需要额外处理。

摘要由CSDN通过智能技术生成

前言

对现有onnx 进行剪切，得到部分onnx结构

Code

import onnx

org_name="new_sim.onnx"
# save_name="./example.onnx"
# save_name="./example2.onnx"
# save_name="./example3.onnx"
save_name="./example4.onnx"


input_name = ["input"]
# output_name = ["81"]
# output_name = ["83"]
# output_name = ["107"]
output_name = ["126"]

onnx.utils.extract_model(org_name, save_name, input_name, output_name)
print("DONE!!!")