Столкновение с проблемой при интеграции пользовательской модели TFLITE с флаттером с помощью плагина tflite - PullRequest
Новая
       7

Столкновение с проблемой при интеграции пользовательской модели TFLITE с флаттером с помощью плагина tflite

1 голос
/ 11 января 2020

Я создал собственную модель с использованием Keras и преобразовал ее в модель TFLITE. Форма ввода для модели CNN - 60 * 60 * 3, так как тип данных int, требуемый буфер будет 60 * 60 * 3 * 4 = 43200. Теперь при интеграции модели с Flutter с использованием функции - 

    recognitions = await Tflite.runModelOnImage(
          path: widget.imagePath,
          imageMean: 127.5,
          imageStd: 127.5,
          numResults: 2,
          threshold: 0.1,
          asynch: true);
      print(recognitions);
    }

я получил эту ошибку - 

    E/AndroidRuntime(21412): Caused by: java.lang.IllegalArgumentException: Cannot convert between a TensorFlowLite buffer with 21600 bytes and a Java Buffer with 43200 bytes.
E/AndroidRuntime(21412):    at org.tensorflow.lite.Tensor.throwIfShapeIsIncompatible(Tensor.java:332)
E/AndroidRuntime(21412):    at org.tensorflow.lite.Tensor.throwIfDataIsIncompatible(Tensor.java:305)
E/AndroidRuntime(21412):    at org.tensorflow.lite.Tensor.setTo(Tensor.java:123)
E/AndroidRuntime(21412):    at org.tensorflow.lite.NativeInterpreterWrapper.run(NativeInterpreterWrapper.java:150)
E/AndroidRuntime(21412):    at org.tensorflow.lite.Interpreter.runForMultipleInputsOutputs(Interpreter.java:311)
E/AndroidRuntime(21412):    at org.tensorflow.lite.Interpreter.run(Interpreter.java:272)
E/AndroidRuntime(21412):    at sq.flutter.tflite.TflitePlugin$RunModelOnImage.runTflite(TflitePlugin.java:452)
E/AndroidRuntime(21412):    at sq.flutter.tflite.TflitePlugin$TfliteTask.doInBackground(TflitePlugin.java:419)
E/AndroidRuntime(21412):    at sq.flutter.tflite.TflitePlugin$TfliteTask.doInBackground(TflitePlugin.java:393)
E/AndroidRuntime(21412):    at android.os.AsyncTask$2.call(AsyncTask.java:333)
E/AndroidRuntime(21412):    at java.util.concurrent.FutureTask.run(FutureTask.java:266)
E/AndroidRuntime(21412):    ... 4 more

Любые предложения о том, как можно решить эту ошибку?

...