Я тренирую модель для классификации изображений в TensorFlow, используя предварительно обученную модель MobileNet и встраивая файл .tflite в приложение Android.
Так получилось, что если я добавлю model.add(layers.Flatten()) после добавления модели MobileNet, при выполнении приложения для Android я получу следующую ошибку: Caused by: java.lang.IllegalArgumentException: ByteBuffer is not a valid flatbuffer model
Однако, если я использую model.add(layers.GlobalAveragePooling2D()) после модели MobileNet приложение работает нормально.
Может ли кто-нибудь помочь мне с объяснением, почему layers.Flatten() вызывает проблемы?
Полный код доступен здесь .