Different shape of input (1, C, H, W)

am15h / tflite_flutter_helper

TensorFlow Lite Flutter Helper Library

https://pub.dev/packages/tflite_flutter_helper

Apache License 2.0

159 stars 284 forks source link

Different shape of input (1, C, H, W) #29

Open longlivelonger opened 3 years ago

longlivelonger commented 3 years ago

Hi, my tflite model converted from Pytorch has input shape [1, C, H, W]. But TensorImage always gives me a tensor with shape [H, W, C]. I just gave up on reading source code. Please could you give me a clue on how to moveaxis or another solution to this case?

InputTensor of model has shape [1, 3, 240, 240] . OutputTensor has shape [1,1]. Its regression task.

SmiffyKMc commented 2 years ago

Have you managed to progress on this?