Expand Dims in Flutter-TfLite for Image Classification

[MinSiThu]

How to expand_dims in flutter tflite

I am making an image classification app.

My tflite model has input shape = (1,244,244,3)
After reading image from image picker, and load as TensorImage, my image shape is (244,244,3).

When I put the image to run in the interpreter

_model.interpreter.run(inputTensorImage.buffer, outputBuffer.buffer);

Following error occurs.
StateError (Bad state: failed precondition)

How can I change my image dimension (244,244,3) into (1,244,244,3) in flutter-tflite?

[gregorscholz]

Just put the input inside [ ].
You can also see this in the given example here.

[MinSiThu]

For other people, who are searching for solution, here is a gist of code.

How to expand dims in flutter-tflite

final imageMatrix = List.generate(
        imageInput.height,
        (y) => List.generate(
          imageInput.width,
          (x) {
            final pixel = imageInput.getPixel(x, y);
            return [pixel.r, pixel.g, pixel.b];
          },
        ),
      );

newMatrix = [imageMatrix];