我有一个基于数据格式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')
问题是:我想在需要数据格式的CPU设备上使用此model.pb。但是,我的模型基于NCHW数据格式。是否有任何方法可以将Model.pb的数据格式从NCHW转换为NHWC?我有一个基于数据格式N(批)x C(通道)x H(高度)x W(宽度)在Pytorch中训练的CNN模型。我将预先训练的模型另存为model.pth。之后,我转换了pre -...NHWC
input.transpose(1,2).transpose(2,3)
吗?