标签： tensorflow onnx
我有一个经过tensorflow训练的模型，冻结到pb文件中。 该模型为NHWC格式。 我的意图是在NCHW中将pb文件转换为ONNX模型。 此链接code is in this answer仅转换为输入图像格式，并且整个模型格式已更改。
但是对我来说，我想拥有NCHW格式的模型，以便将pb文件转换为ONNX时，它将按照NCHW格式保存。
应该在Tensorflow或ONNX中完成吗？
如何转换？还是需要重新训练NCHW图片？