tensorflow - tflite_convert ValueError 未知层 BatchNorm

Question

我们正在使用 Tensorflow 1.14 Keras 2.1.2 GPU：GeForce GTX 1660 Ti 主要：7 次要：5 memoryClockRate(GHz)：1.86 使用来自此 repo https://github.com/matterport/Mask_RCNN的 Mask-RCNN 进行自定义对象检测。现在我们成功地训练了一个模型，并在我们的桌面上检测了它的对象。现在，我们要为tflite面临以下错误的移动使用生成：

ValueError: Unknown layer BatchNorm

请注意，我们.h5使用不同的脚本创建了权重和 keras 模型

我们尝试了以下代码将 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)

标签： tensorflowkerastensorflow-lite