已经转化好的.onnx模型通道顺序是NHWC怎么转换成NCHW 的.onnx 模型 python
时间: 2024-04-01 10:36:04 浏览: 32
可以使用onnx包中的API对.onnx模型进行通道顺序的转换。具体的代码实现如下:
```python
import onnx
from onnx import numpy_helper
# 加载模型
model = onnx.load('model.onnx')
# 获取输入节点和输入张量的信息
input_name = model.graph.input[0].name
input_shape = model.graph.input[0].type.tensor_type.shape.dim
input_dtype = model.graph.input[0].type.tensor_type.elem_type
# 将输入节点的通道顺序从NHWC修改为NCHW
input_shape[1].dim_value, input_shape[2].dim_value, input_shape[3].dim_value = input_shape[3].dim_value, input_shape[1].dim_value, input_shape[2].dim_value
# 创建新的输入张量
new_input_tensor = numpy_helper.from_array(np.zeros([batch_size, channel, height, width], dtype=np.float32), input_name)
# 将新的输入张量添加到模型中
model.graph.input.remove(model.graph.input[0])
model.graph.input.extend([new_input_tensor])
# 将模型保存为新的.onnx文件
onnx.save(model, 'new_model.onnx')
```
其中,我们首先加载了原始的.onnx模型,并获取了输入节点和输入张量的信息。然后,我们将输入节点的通道顺序从NHWC修改为NCHW,并创建了新的输入张量。接着,我们将原始的输入张量从模型中删除,并将新的输入张量添加到模型中。最后,我们将新的模型保存为新的.onnx文件。
相关推荐
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![rar](https://img-home.csdnimg.cn/images/20210720083606.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)