The way to change the data format of pytorch, onnx, or tensorflow (saved_model, frozen model) model from NCHW to NHWC

后端 未结 0 1646
感情败类
感情败类 2020-12-17 10:07

I have my own pytorch, onnx, tf1.x(frozen_graph.pb), and tf2.x(saved_model.pb) models.

Each of those model outputs are exactly same what I wanted. (original model is

相关标签:
回答
  • 消灭零回复
提交回复
热议问题