onnx转tflite 如何让维度保持不变
zhang13359 opened this issue · comments
我用https://netron.app/ 打开model.tflite时,发现里面多了一个Transpose,它将(1,28,28,1)转换为了(1,1,28,28),我并不想让他转换
因为pytorch和onnx数据格式基本都是channel first的,而tensorflow是channel last的。因此你只需要对输出做一个tranpose。
Tool for onnx->keras or onnx->tflite. If tool is useful for you, please star it.
zhang13359 opened this issue · comments
我用https://netron.app/ 打开model.tflite时,发现里面多了一个Transpose,它将(1,28,28,1)转换为了(1,1,28,28),我并不想让他转换
因为pytorch和onnx数据格式基本都是channel first的,而tensorflow是channel last的。因此你只需要对输出做一个tranpose。