Мы используем Tensorflow 1.14 Keras 2.1.2 GPU: GeForce GTX 1660 Ti major: 7 minor: 5 memoryClockRate (GHz): 1.86 для обнаружения пользовательских объектов с использованием Mask-RCNN из этого репозитория https://github.com/matterport/Mask_RCNN. Теперь мы тренировалисьмодель успешно и обнаружение объектов на нашем рабочем столе. Теперь мы хотим сгенерировать tflite для мобильного использования, когда мы сталкиваемся с нижеуказанной ошибкой:
ValueError: Unknown layer BatchNorm
Обратите внимание, что мы создали модель весов и кера .h5 с различными сценариями
Мы попробовали следующий код для преобразования модели keras в tflite
import tensorflow as tf
converter = tf.lite.TFLiteConverter.from_keras_model_file( 'Save-Model8.h5')
tfmodel = converter.convert()
open ("model.tflite", "wb") .write(tfmodel)