onnx模型截取部分

牛andmore牛

于 2024-05-11 14:44:32 发布

阅读量357

点赞数 3

分类专栏： ONNX 文章标签： onnx模型截取

本文链接：https://blog.csdn.net/u011119817/article/details/138715233

版权

ONNX 专栏收录该内容

8 篇文章 1 订阅

订阅专栏

这个是有需求的，比如有多个输入节点，我只用其中几个，或有多个输出节点，我只用其中几个。
在这里插入图片描述

比如这个输入，我们可以直接把transpose去掉，用类pytorch的N,C,H,W的格式输入。
还有如下输出：

transpose，reshpe这种实际都是无效操作，可以放到后处理中。
代码如下：

import onnx  

# onnx.utils.extract_model('segmentation_v2_static_shape.onnx', 'segmentation_v2_static_shape_input.onnx', ['deeplabv3_plus/Conv/Conv2D__6:0'], ['bilinear_upsampling_1'])
onnx.utils.extract_model('segmentation_v2_static_shape.onnx', 'segmentation_v2_static_shape_input_again.onnx', ['deeplabv3_plus/Conv/Conv2D__6:0'], ['Resize__589:0'])