onnx模型截取部分

这个是有需求的，比如有多个输入节点，我只用其中几个，或有多个输出节点，我只用其中几个。


比如这个输入，我们可以直接把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'])