For image classification applications on Android, the input needs to be in the form [1,224,224,3], but .tflite file generated by Hugginface scritp receives it in the form [1,3,224,224].
How to change this?
the model input format in a .tflite file generated by the run_image_classification.py script and
model = TFMobileViTForImageClassification.from_pretrained
and
converter = tf.lite.TFLiteConverter.from_keras_model(model)
this file is “channel first”. How to modify it so that it is “channel last”?
See the shape of the input in the image of the .tflite file
