有什么方法可以将.pb文件的数据格式从NCHW转换为NHWC?

问题描述 投票:2回答:1

我有一个基于数据格式N(批)x C(通道)x H(高度)x W(宽度)]在Pytorch中训练的CNN模型。我将经过预训练的模型另存为model.pth。之后,我使用现有函数从[[model.pth-> model.onnx转换了预训练模型:torch.onnx.export(model, dummy_input, "model.onnx")

然后,我通过下面的模块将此

model.onnx-> model.pb

转换为:import onnx from onnx_tf.backend import prepare model_onnx = onnx.load('model.onnx') tf_rep = prepare(model_onnx) tf_rep.export_graph('model.pb')
问题是:我想在需要

NHWC

数据格式的CPU设备上使用此model.pb。但是,我的模型基于NCHW数据格式。是否有任何方法可以将Model.pb的数据格式从NCHW转换为NHWC?
我有一个基于数据格式N(批)x C(通道)x H(高度)x W(宽度)在Pytorch中训练的CNN模型。我将预先训练的模型另存为model.pth。之后,我转换了pre -...
python-3.x tensorflow pytorch conv-neural-network pre-trained-model
1个回答
0
投票
您能换位输入张量吗?例如input.transpose(1,2).transpose(2,3)吗?
© www.soinside.com 2019 - 2024. All rights reserved.