TFLite Interpreter не может загрузить квантованную модель на Android - PullRequest
0 голосов
/ 06 августа 2020

У меня модель TFLite. Вход модели - изображение 256x192, оно квантовано до 16 бит. Он был квантован с помощью этого конвертера:

converter.optimizations = [tf.lite.Optimize.DEFAULT]
converter.target_spec.supported_types = [tf.float16]
tflite_model = converter.convert()

Я пытаюсь загрузить его в свое приложение android, и при выполнении tflite = new Interpreter(tfliteModel, tfliteOptions); сталкивается со следующей проблемой:

E/AndroidRuntime: FATAL EXCEPTION: CameraBackground
    Process: android.example.com.tflitecamerademo, PID: 7943
    java.lang.IllegalArgumentException: Internal error: Cannot create interpreter: Unimplemented data type FLOAT16 (1) in tensor
    Unimplemented data type FLOAT16 (1) in tensor

Что я могу попробовать, чтобы решить эту проблему?

Спасибо

1 Ответ

0 голосов
/ 09 августа 2020

РЕДАКТИРОВАТЬ: Похоже, вы должны использовать tf.float16, начиная с tf 2 https://www.tensorflow.org/lite/convert/1x_compatibility#unsupported_apis

Может быть проблема в github

...